added missing deps and get tests to work

daisybio · Oct 16, 2024 · 8e0d639 · 8e0d639
1 parent 77a6ca8
commit 8e0d639
Show file tree

Hide file tree

Showing 6 changed files with 1,290 additions and 51 deletions.
diff --git a/drevalpy/models/drp_model.py b/drevalpy/models/drp_model.py
@@ -46,7 +46,7 @@ def get_hyperparameter_set(cls, hyperparameter_file: Optional[str] = None):
             hyperparameter_file = os.path.join(os.path.dirname(inspect.getfile(cls)), "hyperparameters.yaml")
 
         with open(hyperparameter_file, encoding="utf-8") as f:
-            hpams = yaml.safe_load(f, Loader=yaml.FullLoader)[cls.model_name]
+            hpams = yaml.load(f, Loader=yaml.FullLoader)[cls.model_name]
         if hpams is None:
             return [{}]
         # each param should be a list