Expose ABCs for registration.

GAA-UAM · Mar 29, 2023 · 546f360 · 546f360
1 parent 73e7708
commit 546f360
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 3 deletions.
diff --git a/skfda/preprocessing/registration/__init__.py b/skfda/preprocessing/registration/__init__.py
@@ -18,6 +18,10 @@
         "validation",
     ],
     submod_attrs={
+        "_base": [
+            "RegistrationTransformer",
+            "InductiveRegistrationTransformer",
+        ],
         "_fisher_rao": ["ElasticRegistration", "FisherRaoElasticRegistration"],
         "_landmark_registration": [
             "landmark_elastic_registration",
@@ -35,6 +39,10 @@
 )
 
 if TYPE_CHECKING:
+    from ._base import (
+        InductiveRegistrationTransformer as InductiveRegistrationTransformer,
+        RegistrationTransformer as RegistrationTransformer,
+    )
     from ._fisher_rao import (
         ElasticRegistration as ElasticRegistration,
         FisherRaoElasticRegistration as FisherRaoElasticRegistration,

diff --git a/skfda/preprocessing/registration/base.py → skfda/preprocessing/registration/_base.py b/skfda/preprocessing/registration/base.py → skfda/preprocessing/registration/_base.py
diff --git a/skfda/preprocessing/registration/_fisher_rao.py b/skfda/preprocessing/registration/_fisher_rao.py
@@ -16,7 +16,7 @@
 from ...representation.basis import Basis
 from ...representation.interpolation import SplineInterpolation
 from ...typing._numpy import ArrayLike
-from .base import InductiveRegistrationTransformer
+from ._base import InductiveRegistrationTransformer
 
 _MeanType = Callable[[FDataGrid], FDataGrid]
 SelfType = TypeVar("SelfType", bound="FisherRaoElasticRegistration")

diff --git a/skfda/preprocessing/registration/_lstsq_shift_registration.py b/skfda/preprocessing/registration/_lstsq_shift_registration.py
@@ -15,7 +15,7 @@
 from ...representation.extrapolation import ExtrapolationLike
 from ...typing._base import GridPointsLike
 from ...typing._numpy import ArrayLike, NDArrayFloat
-from .base import InductiveRegistrationTransformer
+from ._base import InductiveRegistrationTransformer
 
 SelfType = TypeVar("SelfType", bound="LeastSquaresShiftRegistration[FData]")
 T = TypeVar("T", bound=FData)

diff --git a/skfda/preprocessing/registration/validation.py b/skfda/preprocessing/registration/validation.py
@@ -11,7 +11,7 @@
 from ...misc.validation import check_fdata_dimensions
 from ...representation import FData
 from ...typing._numpy import NDArrayFloat
-from .base import RegistrationTransformer
+from ._base import RegistrationTransformer
 
 Input = TypeVar("Input", bound=FData)
 Output = TypeVar("Output", bound=FData)