diff --git a/test/adaptor/mxnet_adaptor/test_mxnet_query_fwk.py b/test/adaptor/mxnet_adaptor/test_mxnet_query_fwk.py index 7a63e7f60f9..e896e093ec5 100644 --- a/test/adaptor/mxnet_adaptor/test_mxnet_query_fwk.py +++ b/test/adaptor/mxnet_adaptor/test_mxnet_query_fwk.py @@ -30,7 +30,8 @@ def test_one_shot_query(self): def test_get_version(self): Query_version = self.Queryhandler.get_version() - self.assertEqual(mx.__version__, Query_version) + # if the mxnet version not in cfgs, the default maybe be ok. + self.assertNotIn([mx.__version__, 'default'], [Query_version]) def test_get_precisions(self): Query_precisions = self.Queryhandler.get_precisions()