fix

Author: v-shuawei

anomaly-detector/tests/e2e_test_1.py anomaly-detector/tests/e2e_test_mvad.py

@@ -2,10 +2,9 @@
 import numpy as np
 import pandas as pd
 from anomaly_detector import MultivariateAnomalyDetector
-from anomaly_detector import UnivariateAnomalyDetector
 from mlflow.models import infer_signature
 
-mlflow.set_tracking_uri(uri="http://127.0.0.1:5000")
+mlflow.set_tracking_uri(uri="http://127.0.0.1:8080")
 
 
 def main():
@@ -26,7 +25,7 @@ def main():
         model.fit(training_data, params=params)
         predict_params = {"start_time": "string", "end_time": "string"}
         signature = infer_signature(params=predict_params)
-        print(model)
+
         model_info = mlflow.pyfunc.log_model(
             python_model=model,
             artifact_path="mvad_artifacts",
@@ -52,4 +51,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()

anomaly-detector/tests/e2e_test_2.py anomaly-detector/tests/e2e_test_uvad.py

@@ -4,7 +4,7 @@
 from anomaly_detector import EntireAnomalyDetector
 from mlflow.models import infer_signature
 
-mlflow.set_tracking_uri(uri="http://127.0.0.1:5000")
+mlflow.set_tracking_uri(uri="http://127.0.0.1:8080")
 
 
 def main():
@@ -24,7 +24,7 @@ def main():
         model = EntireAnomalyDetector()
 
         signature = infer_signature(params=params)
-        print(model)
+        
         model_info = mlflow.pyfunc.log_model(
             python_model=model,
             artifact_path="uvad_artifacts",