chore: small print

dhis2-chap · Oct 3, 2024 · 5a57b29 · 5a57b29
1 parent a8667c6
commit 5a57b29
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/chap_core/external/mlflow.py b/chap_core/external/mlflow.py
@@ -22,8 +22,7 @@ def __init__(self, model_path):
         self.model_path = model_path
 
     def train(self, train_file_name, model_file_name):
-        # train_file_name = self.model_path / train_file_name
-        # model_file_name = self.model_path / model_file_name
+        logger.info("Training model using MLflow")
         return mlflow.projects.run(
             str(self.model_path),
             entry_point="train",

diff --git a/chap_core/predictor/published_models.py b/chap_core/predictor/published_models.py
@@ -31,7 +31,7 @@ class ExternalModelSpec(ModelSpec):
         period=PeriodType.week,
         description='Weekly Deep Auto Regressive model',
         author='knutdrand',
-        github_link='https://github.com/knutdrand/weekly_ar_model')
+        github_link='https://github.com/knutdrand/weekly_ar_model'),
 )
 
 model_dict = {model.name: model for model in models}