refactor(io): switch core reader imports to canonical exports

Author: ErjieWu

deepks/core/ml/eval/test.py

@@ -9,7 +9,7 @@
     sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../../")
 from deepks.default import DEVICE
 from deepks.core.ml.models.corrnet import CorrNet
-from deepks.io.readers.group_reader import GroupReader
+from deepks.io.readers import GroupReader
 from deepks.utils import load_yaml, load_dirs, check_list
 
 

deepks/core/ml/train/train.py

@@ -10,7 +10,7 @@
     sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../../")
 from deepks.default import DEVICE
 from deepks.core.ml.models.corrnet import CorrNet
-from deepks.io.readers.group_reader import GroupReader
+from deepks.io.readers import GroupReader
 from deepks.utils import load_dirs, load_elem_table
 from deepks.core.ml.utils import preprocess, fit_elem_const, make_loss
 from deepks.core.ml.eval.evaluator import Evaluator, NatomLossList

deepks/model/reader.py

@@ -1,3 +1,5 @@
 """Compatibility shim for the legacy model reader module."""
 
-from deepks.io.readers.group_reader import *  # noqa: F401,F403
+from deepks.io.readers import GroupReader, Reader, SimpleReader
+
+__all__ = ["Reader", "GroupReader", "SimpleReader"]

tests/unit/test_reader_export_compat.py

@@ -0,0 +1,27 @@
+"""
+Reader exports compatibility coverage.
+
+Tests:
+- `test_reader_export_identity`
+"""
+
+from deepks.io.readers import GroupReader as GroupReaderPkg
+from deepks.io.readers import Reader as ReaderPkg
+from deepks.io.readers import SimpleReader as SimpleReaderPkg
+from deepks.io.readers.group_reader import GroupReader as GroupReaderLegacy
+from deepks.io.readers.group_reader import Reader as ReaderLegacy
+from deepks.io.readers.group_reader import SimpleReader as SimpleReaderLegacy
+from deepks.model.reader import GroupReader as GroupReaderModelShim
+from deepks.model.reader import Reader as ReaderModelShim
+from deepks.model.reader import SimpleReader as SimpleReaderModelShim
+
+
+def test_reader_export_identity():
+    """Canonical exports and compatibility shims should point to identical class objects."""
+    assert ReaderPkg is ReaderLegacy
+    assert GroupReaderPkg is GroupReaderLegacy
+    assert SimpleReaderPkg is SimpleReaderLegacy
+
+    assert ReaderPkg is ReaderModelShim
+    assert GroupReaderPkg is GroupReaderModelShim
+    assert SimpleReaderPkg is SimpleReaderModelShim