fix ext

Author: xadupre

onnx_diagnostic/_command_lines_parser.py

@@ -316,7 +316,11 @@ def _cmd_validate(argv: List[Any]):
             print(f"  + {k.ljust(max_length)}: {_ds_clean(v)}")
     else:
         # Let's skip any invalid combination if known to be unsupported
-        if "onnx" not in args.export and "custom" not in args.export and args.opt:
+        if (
+            "onnx" not in (args.export or "")
+            and "custom" not in (args.export or "")
+            and (args.opt or "")
+        ):
             print(f"validate - unsupported args: export={args.export!r}, opt={args.opt!r}")
             return
         summary, _data = validate_model(

onnx_diagnostic/ext_test_case.py

@@ -934,6 +934,8 @@ def assertEqualAny(
         elif expected.__class__.__name__ in ("Dim", "_Dim", "_DimHintType"):
             self.assertEqual(type(expected), type(value), msg=msg)
             self.assertEqual(expected.__name__, value.__name__, msg=msg)
+        elif expected is None:
+            self.assertEqual(expected, value, msg=msg)
         else:
             raise AssertionError(
                 f"Comparison not implemented for types {type(expected)} and {type(value)}"