[ENH] Allow multiple instances from multiple mock classes in _safe_import (sktime#1818)

The issue was spotted in sktime via
sktime/sktime#8061

Updated the implementation and copied the unit tests to ensure that it
works with pytorch-forecasting.
closes sktime#1815

Author: fnhirwa

pytorch_forecasting/utils/_dependencies/_safe_import.py

@@ -1,6 +1,8 @@
 """Import a module/class, return a Mock object if import fails.
 
 Copied from sktime/skbase.
+
+Should be refactored and moved to a common location in skbase.
 """
 
 import importlib
@@ -12,20 +14,21 @@
 def _safe_import(import_path, pkg_name=None):
     """Import a module/class, return a Mock object if import fails.
 
+    Idiomatic usage is ``obj = _safe_import("a.b.c.obj")``.
     The function supports importing both top-level modules and nested attributes:
 
-    - Top-level module: "torch" -> imports torch
-    - Nested module: "torch.nn" -> imports torch.nn
-    - Class/function: "torch.nn.Linear" -> imports Linear class from torch.nn
+    - Top-level module: ``"torch"`` -> same as ``import torch``
+    - Nested module: ``"torch.nn"`` -> same as``from torch import nn``
+    - Class/function: ``"torch.nn.Linear"`` -> same as ``from torch.nn import Linear``
 
     Parameters
     ----------
     import_path : str
         The path to the module/class to import. Can be:
 
-        - Single module: "torch"
-        - Nested module: "torch.nn"
-        - Class/attribute: "torch.nn.ReLU"
+        - Single module: ``"torch"``
+        - Nested module: ``"torch.nn"``
+        - Class/attribute: ``"torch.nn.ReLU"``
 
         Note: The dots in the path determine the import behavior:
 
@@ -37,39 +40,44 @@ def _safe_import(import_path, pkg_name=None):
         The name of the package to check for installation. This is useful when
         the import name differs from the package name, for example:
 
-        - import: "sklearn" -> pkg_name="scikit-learn"
-        - import: "cv2" -> pkg_name="opencv-python"
+        - import: ``"sklearn"`` -> ``pkg_name="scikit-learn"``
+        - import: ``"cv2"`` -> ``pkg_name="opencv-python"``
 
-        If None, uses the first part of import_path before the dot.
+        If ``None``, uses the first part of ``import_path`` before the dot.
 
     Returns
     -------
     object
-        One of the following:
+        If the import path and ``pkg_name`` is present, one of the following:
+
+        - The imported module if ``import_path`` has no dots
+        - The imported submodule if ``import_path`` has one dot
+        - The imported class/function if ``import_path`` has multiple dots
 
-        - The imported module if import_path has no dots
-        - The imported submodule if import_path has one dot
-        - The imported class/function if import_path has multiple dots
-        - A MagicMock object that returns an installation message if the
-          package is not found
+        If the package or import path are not found:
+        a unique ``MagicMock`` object per unique import path.
 
     Examples
     --------
+    >>> from pytorch_forecasting.utils.dependencies._safe_import import _safe_import
+
     >>> # Import a top-level module
-    >>> torch = safe_import("torch")
+    >>> torch = _safe_import("torch")
 
     >>> # Import a submodule
-    >>> nn = safe_import("torch.nn")
+    >>> nn = _safe_import("torch.nn")
 
     >>> # Import a specific class
-    >>> Linear = safe_import("torch.nn.Linear")
+    >>> Linear = _safe_import("torch.nn.Linear")
 
     >>> # Import with different package name
-    >>> cv2 = safe_import("cv2", pkg_name="opencv-python")
+    >>> cv2 = _safe_import("cv2", pkg_name="opencv-python")
     """
+    path_list = import_path.split(".")
+
     if pkg_name is None:
-        path_list = import_path.split(".")
         pkg_name = path_list[0]
+    obj_name = path_list[-1]
 
     if pkg_name in _get_installed_packages():
         try:
@@ -79,10 +87,44 @@ def _safe_import(import_path, pkg_name=None):
             module = importlib.import_module(module_name)
             return getattr(module, attr_name)
         except (ImportError, AttributeError):
-            return importlib.import_module(import_path)
-    else:
-        mock_obj = MagicMock()
-        mock_obj.__call__ = MagicMock(
-            return_value=f"Please install {pkg_name} to use this functionality."
-        )
-        return mock_obj
+            pass
+
+    mock_obj = _create_mock_class(obj_name)
+    return mock_obj
+
+
+class CommonMagicMeta(type):
+    def __getattr__(cls, name):
+        return MagicMock()
+
+    def __setattr__(cls, name, value):
+        pass  # Ignore attribute writes
+
+
+class MagicAttribute(metaclass=CommonMagicMeta):
+    def __getattr__(self, name):
+        return MagicMock()
+
+    def __setattr__(self, name, value):
+        pass  # Ignore attribute writes
+
+    def __call__(self, *args, **kwargs):
+        return self  # Ensures instantiation returns the same object
+
+
+def _create_mock_class(name: str, bases=()):
+    """Create new dynamic mock class similar to MagicMock.
+
+    Parameters
+    ----------
+    name : str
+        The name of the new class.
+    bases : tuple, default=()
+        The base classes of the new class.
+
+    Returns
+    -------
+    a new class that behaves like MagicMock, with name ``name``.
+        Forwards all attribute access to a MagicMock object stored in the instance.
+    """
+    return type(name, (MagicAttribute,), {"__metaclass__": CommonMagicMeta})

pytorch_forecasting/utils/_dependencies/tests/__init__.py

@@ -0,0 +1 @@
+"""Tests for dependency utilities."""

pytorch_forecasting/utils/_dependencies/tests/test_safe_import.py

@@ -0,0 +1,76 @@
+__author__ = ["jgyasu", "fkiraly"]
+
+from pytorch_forecasting.utils._dependencies import (
+    _get_installed_packages,
+    _safe_import,
+)
+
+
+def test_import_present_module():
+    """Test importing a dependency that is installed."""
+    result = _safe_import("pandas")
+    assert result is not None
+    assert "pandas" in _get_installed_packages()
+
+
+def test_import_missing_module():
+    """Test importing a dependency that is not installed."""
+    result = _safe_import("nonexistent_module")
+    assert hasattr(result, "__name__")
+    assert result.__name__ == "nonexistent_module"
+
+
+def test_import_without_pkg_name():
+    """Test importing a dependency with the same name as package name."""
+    result = _safe_import("torch", pkg_name="torch")
+    assert result is not None
+
+
+def test_import_with_different_pkg_name_1():
+    """Test importing a dependency with a different package name."""
+    result = _safe_import("skbase", pkg_name="scikit-base")
+    assert result is not None
+
+
+def test_import_with_different_pkg_name_2():
+    """Test importing another dependency with a different package name."""
+    result = _safe_import("cv2", pkg_name="opencv-python")
+    assert result is not None
+
+
+def test_import_submodule():
+    """Test importing a submodule."""
+    result = _safe_import("torch.nn")
+    assert result is not None
+
+
+def test_import_class():
+    """Test importing a class."""
+    result = _safe_import("torch.nn.Linear")
+    assert result is not None
+
+
+def test_import_existing_object():
+    """Test importing an existing object."""
+    result = _safe_import("pandas.DataFrame")
+    assert result is not None
+    assert result.__name__ == "DataFrame"
+    from pandas import DataFrame
+
+    assert result is DataFrame
+
+
+def test_multiple_inheritance_from_mock():
+    """Test multiple inheritance from dynamic MagicMock."""
+    Class1 = _safe_import("foobar.foo.FooBar")
+    Class2 = _safe_import("barfoobar.BarFooBar")
+
+    class NewClass(Class1, Class2):
+        """This should not trigger an error.
+
+        The class definition would trigger an error if multiple inheritance
+        from Class1 and Class2 does not work, e.g., if it is simply
+        identical to MagicMock.
+        """
+
+        pass

tests/test_utils/test_safe_import.py

@@ -0,0 +1,70 @@
+from pytorch_forecasting.utils._dependencies import _safe_import
+
+
+def test_present_module():
+    """Test importing a dependency that is installed."""
+    module = _safe_import("torch")
+    assert module is not None
+
+
+def test_import_missing_module():
+    """Test importing a dependency that is not installed."""
+    result = _safe_import("nonexistent_module")
+    assert hasattr(result, "__name__")
+    assert result.__name__ == "nonexistent_module"
+
+
+def test_import_without_pkg_name():
+    """Test importing a dependency with the same name as package name."""
+    result = _safe_import("torch", pkg_name="torch")
+    assert result is not None
+
+
+def test_import_with_different_pkg_name_1():
+    """Test importing a dependency with a different package name."""
+    result = _safe_import("skbase", pkg_name="scikit-base")
+    assert result is not None
+
+
+def test_import_with_different_pkg_name_2():
+    """Test importing another dependency with a different package name."""
+    result = _safe_import("cv2", pkg_name="opencv-python")
+    assert result is not None
+
+
+def test_import_submodule():
+    """Test importing a submodule."""
+    result = _safe_import("torch.nn")
+    assert result is not None
+
+
+def test_import_class():
+    """Test importing a class."""
+    result = _safe_import("torch.nn.Linear")
+    assert result is not None
+
+
+def test_import_existing_object():
+    """Test importing an existing object."""
+    result = _safe_import("pandas.DataFrame")
+    assert result is not None
+    assert result.__name__ == "DataFrame"
+    from pandas import DataFrame
+
+    assert result is DataFrame
+
+
+def test_multiple_inheritance_from_mock():
+    """Test multiple inheritance from dynamic MagicMock."""
+    Class1 = _safe_import("foobar.foo.FooBar")
+    Class2 = _safe_import("barfoobar.BarFooBar")
+
+    class NewClass(Class1, Class2):
+        """This should not trigger an error.
+
+        The class definition would trigger an error if multiple inheritance
+        from Class1 and Class2 does not work, e.g., if it is simply
+        identical to MagicMock.
+        """
+
+        pass