Fix broken type hints

openml · Aug 28, 2024 · 18f879d · 18f879d
1 parent 1a04502
commit 18f879d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/amlb/results.py b/amlb/results.py
@@ -12,7 +12,7 @@
 import os
 import re
 import statistics
-from typing import Union
+from typing import Union, cast
 
 import numpy as np
 from numpy import nan, sort
@@ -297,7 +297,7 @@ def save_predictions(dataset: Dataset, output_file: str,
         if isinstance(predictions, S):
             predictions = predictions.values
         if scipy.sparse.issparse(truth) and truth.shape[1] == 1:
-            truth = pd.DataFrame(truth.todense())
+            truth = pd.DataFrame(cast(scipy.sparse.sparray, truth).todense())
         if isinstance(truth, DF):
             truth = truth.squeeze()
         if isinstance(truth, S):

diff --git a/amlb/utils/config.py b/amlb/utils/config.py
@@ -42,7 +42,7 @@ def yaml_load(file, as_namespace=False):
             yaml = YAML(typ='safe')
         return yaml.load(file)
 else:
-    def yaml_load(*_, **__):
+    def yaml_load(*_, **__):  # type: ignore[misc]
         raise ImportError("ruamel.yaml package is required to load `yaml` config files.")