Add params to model_evaluate for logging

DataScientest-Studio · Nov 12, 2024 · 6b3ddd2 · 6b3ddd2
1 parent 473988f
commit 6b3ddd2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/config_manager.py b/src/config_manager.py
@@ -123,6 +123,7 @@ def get_model_evaluation_config(self) -> ModelEvaluationConfig:
         repo_owner = "castolpe"
         repo_name = "sep24_bmlops_int_reco_films"
         repo_token = os.getenv("DAGSHUB_TOKEN")
+        params = self.params.NearestNeighbors
 
         create_directories([config.root_dir])
 
@@ -135,6 +136,7 @@ def get_model_evaluation_config(self) -> ModelEvaluationConfig:
             repo_name=repo_name,
             repo_token=repo_token,
             mlflow_uri=f"https://dagshub.com/{repo_owner}/{repo_name}.mlflow",
+            params=params,
         )
 
         return model_evaluation_config
diff --git a/src/models_module_def/model_evaluate.py b/src/models_module_def/model_evaluate.py
@@ -23,8 +23,6 @@ class ModelEvaluation:
     def __init__(self, config: ModelEvaluationConfig):
         self.config = config
 
-        print(self.config.repo_token)
-
         dagshub.auth.add_app_token(self.config.repo_token)
 
         dagshub.init(