diff --git a/ngclearn/modules/__init__.py b/ngclearn/modules/__init__.py index 8769469b..b18f84b7 100644 --- a/ngclearn/modules/__init__.py +++ b/ngclearn/modules/__init__.py @@ -1,6 +1,6 @@ -from regression.elastic_net import Iterative_ElasticNet -from regression.lasso import Iterative_Lasso -from regression.ridge import Iterative_Ridge +from .regression.elastic_net import Iterative_ElasticNet +from .regression.lasso import Iterative_Lasso +from .regression.ridge import Iterative_Ridge diff --git a/ngclearn/modules/regression/__init__.py b/ngclearn/modules/regression/__init__.py index 2d7ce064..064d5303 100644 --- a/ngclearn/modules/regression/__init__.py +++ b/ngclearn/modules/regression/__init__.py @@ -1,6 +1,6 @@ -from elastic_net import Iterative_ElasticNet -from lasso import Iterative_Lasso -from ridge import Iterative_Ridge +from .elastic_net import Iterative_ElasticNet +from .lasso import Iterative_Lasso +from .ridge import Iterative_Ridge