refactor(compat): convert model shims to lazy module forwarding

Author: ErjieWu

deepks/model/evaluator.py

@@ -1,3 +1,17 @@
 """Compatibility shim for the legacy evaluator module."""
 
-from deepks.core.ml.eval.evaluator import *  # noqa: F401,F403
+import importlib
+
+
+_impl = importlib.import_module("deepks.core.ml.eval.evaluator")
+
+
+def __getattr__(name):
+	return getattr(_impl, name)
+
+
+def __dir__():
+	return sorted(set(globals()) | set(__all__))
+
+
+__all__ = [name for name in dir(_impl) if not name.startswith("_")]

deepks/model/model.py

@@ -1,3 +1,17 @@
 """Compatibility shim for the legacy model module."""
 
-from deepks.core.ml.models.corrnet import *  # noqa: F401,F403
+import importlib
+
+
+_impl = importlib.import_module("deepks.core.ml.models.corrnet")
+
+
+def __getattr__(name):
+	return getattr(_impl, name)
+
+
+def __dir__():
+	return sorted(set(globals()) | set(__all__))
+
+
+__all__ = [name for name in dir(_impl) if not name.startswith("_")]

deepks/model/test.py

@@ -1,3 +1,17 @@
 """Compatibility shim for the legacy model test module."""
 
-from deepks.core.ml.eval.test import *  # noqa: F401,F403
+import importlib
+
+
+_impl = importlib.import_module("deepks.core.ml.eval.test")
+
+
+def __getattr__(name):
+	return getattr(_impl, name)
+
+
+def __dir__():
+	return sorted(set(globals()) | set(__all__))
+
+
+__all__ = [name for name in dir(_impl) if not name.startswith("_")]

deepks/model/train.py

@@ -1,3 +1,17 @@
 """Compatibility shim for the legacy model train module."""
 
-from deepks.core.ml.train.train import *  # noqa: F401,F403
+import importlib
+
+
+_impl = importlib.import_module("deepks.core.ml.train.train")
+
+
+def __getattr__(name):
+	return getattr(_impl, name)
+
+
+def __dir__():
+	return sorted(set(globals()) | set(__all__))
+
+
+__all__ = [name for name in dir(_impl) if not name.startswith("_")]

deepks/model/utils.py

@@ -1,3 +1,17 @@
 """Compatibility shim for the legacy model utils module."""
 
-from deepks.core.ml.utils import *  # noqa: F401,F403
+import importlib
+
+
+_impl = importlib.import_module("deepks.core.ml.utils")
+
+
+def __getattr__(name):
+	return getattr(_impl, name)
+
+
+def __dir__():
+	return sorted(set(globals()) | set(__all__))
+
+
+__all__ = [name for name in dir(_impl) if not name.startswith("_")]

tests/unit/test_model_shim_lazy_exports.py

@@ -0,0 +1,35 @@
+"""
+Legacy model shim export coverage.
+
+Tests:
+- `test_model_train_test_shim_exports`
+- `test_model_eval_utils_shim_exports`
+"""
+
+import importlib
+
+import deepks.model.evaluator as evaluator_shim
+import deepks.model.model as model_shim
+import deepks.model.test as test_shim
+import deepks.model.train as train_shim
+import deepks.model.utils as utils_shim
+
+evaluator_impl = importlib.import_module("deepks.core.ml.eval.evaluator")
+eval_test_impl = importlib.import_module("deepks.core.ml.eval.test")
+model_impl = importlib.import_module("deepks.core.ml.models.corrnet")
+utils_impl = importlib.import_module("deepks.core.ml.utils")
+train_impl = importlib.import_module("deepks.core.ml.train.train")
+
+
+def test_model_train_test_shim_exports():
+    assert train_shim.main is train_impl.main
+    assert train_shim.train is train_impl.train
+    assert test_shim.main is eval_test_impl.main
+    assert test_shim.test is eval_test_impl.test
+
+
+def test_model_eval_utils_shim_exports():
+    assert evaluator_shim.Evaluator is evaluator_impl.Evaluator
+    assert model_shim.CorrNet is model_impl.CorrNet
+    assert utils_shim.fit_elem_const is utils_impl.fit_elem_const
+    assert utils_shim.preprocess is utils_impl.preprocess