mxnet use default version cfg. (#1247)

intel · Sep 15, 2022 · b75ff27 · b75ff27
1 parent 98019c6
commit b75ff27
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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()