update import

azuur · Feb 5, 2024 · acbebd4 · acbebd4
1 parent 71a4e1e
commit acbebd4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ml_pipelines/deployment/aws/serve.py b/ml_pipelines/deployment/aws/serve.py
@@ -7,13 +7,13 @@
 
 import typer
 
-from ml_pipelines.deployment.common.serve import run_serve
-from ml_pipelines.deployment.local.io import (
+from ml_pipelines.deployment.aws.io import (
     get_best_version,
     get_latest_version,
     get_train_artifacts,
     prediction_logging_func,
 )
+from ml_pipelines.deployment.common.serve import run_serve
 
 
 def main(