oserror

Author: xadupre

_unittests/ut_torch_models/test_hghub_api.py

@@ -3,6 +3,7 @@
 from onnx_diagnostic.ext_test_case import (
     ExtTestCase,
     hide_stdout,
+    ignore_errors,
     long_test,
     never_test,
     requires_torch,
@@ -26,6 +27,7 @@ class TestHuggingFaceHubApi(ExtTestCase):
 
     @requires_transformers("4.50")  # we limit to some versions of the CI
     @requires_torch("2.7")
+    @ignore_errors(OSError)  # connectivity issues
     @hide_stdout()
     def test_enumerate_model_list(self):
         models = list(
@@ -76,6 +78,7 @@ def test_get_pretrained_config(self):
 
     @requires_transformers("4.50")
     @requires_torch("2.7")
+    @ignore_errors(OSError)  # connectivity issues
     @hide_stdout()
     def test_get_model_info(self):
         info = get_model_info("microsoft/phi-2")

_unittests/ut_xrun_doc/test_documentation_examples.py

@@ -5,7 +5,13 @@
 import subprocess
 import time
 from onnx_diagnostic import __file__ as onnx_diagnostic_file
-from onnx_diagnostic.ext_test_case import ExtTestCase, is_windows, has_transformers, has_torch
+from onnx_diagnostic.ext_test_case import (
+    ExtTestCase,
+    is_windows,
+    has_transformers,
+    has_torch,
+    ignore_errors,
+)
 
 
 VERBOSE = 0
@@ -94,6 +100,7 @@ def _test_(self, name=name):
 
             else:
 
+                @ignore_errors(OSError)  # connectivity issues
                 def _test_(self, name=name):
                     res = self.run_test(fold, name, verbose=VERBOSE)
                     self.assertTrue(res)