diff --git a/src/config_manager.py b/src/config_manager.py index cf94578..777b186 100644 --- 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 index 2cafb43..23dfb0a 100644 --- 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(