typing: Fix mypy errors split.py

openml · LennartPurucker · Jan 9, 2024 · Jan 8, 2024 · Jan 8, 2024 · Jan 8, 2024
commit 053b0532bd4619e610e4d9dbafd2ad89e2f4b651
diff --git a/openml/tasks/split.py b/openml/tasks/split.py
@@ -4,9 +4,10 @@
 import pickle
 from collections import OrderedDict
 from pathlib import Path
+from typing import Any
 from typing_extensions import NamedTuple
 
-import arff
+import arff  # type: ignore
 import numpy as np
 
 
@@ -55,7 +56,7 @@ def __init__(
         self.folds = len(self.split[0])
         self.samples = len(self.split[0][0])
 
-    def __eq__(self, other):
+    def __eq__(self, other: Any) -> bool:
         if (
             (not isinstance(self, type(other)))
             or self.name != other.name