diff --git a/AtlasMl/atlasml/ml/__init__.py b/AtlasMl/atlasml/ml/__init__.py index 6ed9ce0..8b13789 100644 --- a/AtlasMl/atlasml/ml/__init__.py +++ b/AtlasMl/atlasml/ml/__init__.py @@ -1,4 +1 @@ -from .ml_runner import ml_runner -from .model import ml_model_handler -__all__ = ["ml_runner", "ml_model_handler"] diff --git a/AtlasMl/atlasml/ml/ml_runner.py b/AtlasMl/atlasml/ml/ml_runner.py deleted file mode 100644 index 158de31..0000000 --- a/AtlasMl/atlasml/ml/ml_runner.py +++ /dev/null @@ -1,11 +0,0 @@ -import logging - -from atlasml.ml.model import ml_model_handler - -def ml_runner(): - print("Running ML model") - print(ml_model_handler()) - - -if __name__ == "__main__": - ml_runner() \ No newline at end of file diff --git a/AtlasMl/atlasml/ml/model.py b/AtlasMl/atlasml/ml/model.py deleted file mode 100644 index c6f2522..0000000 --- a/AtlasMl/atlasml/ml/model.py +++ /dev/null @@ -1,3 +0,0 @@ - -def ml_model_handler(): - return {"status": "ok"} \ No newline at end of file