Fixed bugs

adnaniazi · Jul 10, 2024 · ce0dbf4 · ce0dbf4
1 parent 95c2e2f
commit ce0dbf4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -54,4 +54,3 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 [0.1.2]: https://github.com/adnaniazi/capfinder/compare/0.1.1...0.1.2
 [0.1.1]: https://pypi.org/manage/project/capfinder/release/0.1.1/
 [0.1.0]: https://pypi.org/manage/project/capfinder/release/0.1.0/
-
diff --git a/src/capfinder/training.py b/src/capfinder/training.py
@@ -458,7 +458,9 @@ def run_training_pipeline(
     )
     tune_experiment_url = tune_experiment.url
     if model_type not in ["cnn_lstm", "encoder", "resnet"]:
-        raise ValueError("Invalid model type. Expected 'cnn_lstm' or 'encoder'.")
+        raise ValueError(
+            "Invalid model type. Expected 'cnn_lstm' or 'encoder' or 'renset'."
+        )
     model = get_model(model_type)
 
     hyper_model = model(