fix patching module that doesn't exist

src/datasets/utils/patching.py

@@ -53,7 +53,10 @@ def __enter__(self):
         # in this case we need to patch "os" and "os.path"
 
         for i in range(len(submodules)):
-            submodule = import_module(".".join(submodules[: i + 1]))
+            try:
+                submodule = import_module(".".join(submodules[: i + 1]))
+            except ModuleNotFoundError:
+                continue
             # We iterate over all the globals in self.obj in case we find "os" or "os.path"
             for attr in self.obj.__dir__():
                 obj_attr = getattr(self.obj, attr)
@@ -79,7 +82,10 @@ def __enter__(self):
         # itself if it was imported as "from os.path import join".
 
         if submodules:  # if it's an attribute of a submodule like "os.path.join"
-            attr_value = getattr(import_module(".".join(submodules)), target_attr)
+            try:
+                attr_value = getattr(import_module(".".join(submodules)), target_attr)
+            except (AttributeError, ModuleNotFoundError):
+                return
             # We iterate over all the globals in self.obj in case we find "os.path.join"
             for attr in self.obj.__dir__():
                 # We don't check for the name of the global, but rather if its value *is* "os.path.join".

tests/test_patching.py

@@ -142,3 +142,11 @@ def test_patch_submodule_successive():
     assert _test_patching.os.path.join is original_join
     assert _test_patching.os.path.dirname is original_dirname
     assert _test_patching.os.rename is original_rename
+
+
+def test_patch_submodule_doesnt_exist():
+    mock = "__test_patch_submodule_doesnt_exist_mock__"
+    with patch_submodule(_test_patching, "__module_that_doesn_exist__.__attribute_that_doesn_exist__", mock):
+        pass
+    with patch_submodule(_test_patching, "os.__attribute_that_doesn_exist__", mock):
+        pass