disable test using matplotlib

Author: xadupre

_unittests/ut_xrun_doc/test_command_lines_exe.py

@@ -2,6 +2,7 @@
 import unittest
 from contextlib import redirect_stdout
 from io import StringIO
+import torch
 from onnx_diagnostic.ext_test_case import ExtTestCase, ignore_warnings
 from onnx_diagnostic._command_lines_parser import main
 from onnx_diagnostic.helpers.log_helper import enumerate_csv_files
@@ -23,6 +24,9 @@ def test_a_parser_print(self):
                 text = st.getvalue()
                 self.assertIn("Add", text)
 
+    @unittest.skipIf(
+        torch.__version__.startswith("2.9.0"), "Possibly one issue with matplotlib."
+    )
     def test_b_parser_stats(self):
         output = self.get_dump_file("test_parser_stats.xlsx")
         st = StringIO()
@@ -68,6 +72,9 @@ def test_f_parser_validate(self):
         self.assertIn("model_clas", text)
 
     @ignore_warnings(UserWarning)
+    @unittest.skipIf(
+        torch.__version__.startswith("2.9.0"), "Possibly one issue with matplotlib."
+    )
     def test_g_parser_agg(self):
         path = os.path.abspath(
             os.path.join(os.path.dirname(__file__), "..", "ut_helpers", "data")

_unittests/ut_xrun_doc/test_documentation_examples.py

@@ -131,7 +131,7 @@ def add_test_methods(cls):
             }:
                 reason = "too long"
 
-            if not reason and torch.__version__ == "2.9.0":
+            if not reason and torch.__version__.startswith("2.9.0"):
                 reason = "examples are failing for on CI for 2.9.0"
 
             if reason:

_unittests/ut_xrun_doc/test_documentation_technical.py

@@ -77,7 +77,7 @@ def add_test_methods(cls):
             if not reason and not has_dot and name in {"plot_layer_norm_discrepancies.py"}:
                 reason = "dot not installed"
 
-            if not reason and torch.__version__ == "2.9.0":
+            if not reason and torch.__version__.startswith("2.9.0"):
                 reason = "examples are failing for on CI for 2.9.0"
 
             if reason: