test the random forest model in the main pipeline

udacity · Nov 21, 2023 · e16f893 · e16f893
1 parent 8ebd294
commit e16f893
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 10 deletions.
diff --git a/components/test_regression_model/conda.yml b/components/test_regression_model/conda.yml
@@ -4,9 +4,11 @@ channels:
   - defaults
 dependencies:
   - pandas=1.1.4
+  - numpy==1.24.4
   - pip=20.3.3
-  - mlflow=1.14.1
   - scikit-learn=0.24.1
   - pip:
-      - wandb==0.10.31
+      - protobuf==3.20.*
+      - mlflow==2.2.2
+      - wandb==0.12.17
       - git+https://github.com/udacity/nd0821-c2-build-model-workflow-starter.git#egg=wandb-utils&subdirectory=components
diff --git a/components/test_regression_model/run.py b/components/test_regression_model/run.py
@@ -30,7 +30,7 @@ def go(args):
     test_dataset_path = run.use_artifact(args.test_dataset).file()
 
     # Read test dataset
-    X_test = pd.read_csv(test_dataset_path)
+    X_test = pd.read_parquet(test_dataset_path)
     y_test = X_test.pop("price")
 
     logger.info("Loading model and performing inference on test set")

diff --git a/main.py b/main.py
@@ -113,13 +113,12 @@ def go(config: DictConfig):
 
 
         if "test_regression_model" in active_steps:
-
-            ##################
-            # Implement here #
-            ##################
-
-            pass
-
+            _ = mlflow.run (os.path.join(root_path, f"{config['main']['components_repository']}/test_regression_model"),
+                            'main',
+                            parameters={
+                                "mlflow_model": config['modeling']['output_artifact'] +':prod',
+                                "test_dataset": 'test_data.parquet:latest'
+                            })
 
 if __name__ == "__main__":
     go()