opt imports

Author: g0bel1n

.gitignore

@@ -7,4 +7,5 @@ build
 .vscode
 .idea
 .DS_Store
-__pychache__
+__pychache__
+.pytest_cache

TinyAutoML/Models/__init__.py

@@ -1,9 +1,9 @@
-from TinyAutoML.Models.DemocraticModel import DemocraticModel
-from TinyAutoML.Models.BestModel import BestModel
-from TinyAutoML.Models.OneRulerForAll import OneRulerForAll
+from .DemocraticModel import DemocraticModel
+from .BestModel import BestModel
+from .OneRulerForAll import OneRulerForAll
 
-from TinyAutoML.Models.DemocraticModelCV import DemocraticModelCV
-from TinyAutoML.Models.BestModelCV import BestModelCV
-from TinyAutoML.Models.OneRulerForAllCV import OneRulerForAllCV
+from .DemocraticModelCV import DemocraticModelCV
+from .BestModelCV import BestModelCV
+from .OneRulerForAllCV import OneRulerForAllCV
 
 __all__ = ['DemocraticModel','BestModel','OneRulerForAll', 'DemocraticModelCV', 'BestModelCV', 'OneRulerForAllCV']

TinyAutoML/__init__.py

@@ -1,5 +1,5 @@
-from TinyAutoML.Estimator import MetaPipeline
-from TinyAutoML.EstimatorCV import MetaPipelineCV
+from .Estimator import MetaPipeline
+from .EstimatorCV import MetaPipelineCV
 
 
 __all__ = ['MetaPipeline', 'MetaPipelineCV']