Fix invalid type hints for pd.Series

NorskRegnesentral · Aug 21, 2024 · 3c8603e · 3c8603e
1 parent 42cc9c7
commit 3c8603e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/skchange/anomaly_detectors/base.py b/skchange/anomaly_detectors/base.py
@@ -32,7 +32,7 @@ class PointAnomalyDetector(BaseDetector):
     """
 
     @staticmethod
-    def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series[int]:
+    def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series:
         """Convert the sparse output from the predict method to a dense format.
 
         Parameters
@@ -51,7 +51,7 @@ def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series[int]:
         return y_dense
 
     @staticmethod
-    def dense_to_sparse(y_dense: pd.Series) -> pd.Series[int]:
+    def dense_to_sparse(y_dense: pd.Series) -> pd.Series:
         """Convert the dense output from the transform method to a sparse format.
 
         Parameters
@@ -96,9 +96,7 @@ class CollectiveAnomalyDetector(BaseDetector):
     """
 
     @staticmethod
-    def sparse_to_dense(
-        y_sparse: pd.Series[pd.Interval], index: pd.Index
-    ) -> pd.Series[int]:
+    def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series:
         """Convert the sparse output from the predict method to a dense format.
 
         Parameters
@@ -122,7 +120,7 @@ def sparse_to_dense(
         return pd.Series(y_dense, index=index, name="anomaly", dtype="int64")
 
     @staticmethod
-    def dense_to_sparse(y_dense: pd.Series) -> pd.Series[pd.Interval]:
+    def dense_to_sparse(y_dense: pd.Series) -> pd.Series:
         """Convert the dense output from the transform method to a sparse format.
 
         Parameters

diff --git a/skchange/change_detectors/base.py b/skchange/change_detectors/base.py
@@ -35,7 +35,7 @@ class ChangepointDetector(BaseDetector):
     """
 
     @staticmethod
-    def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series[int]:
+    def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series:
         """Convert the sparse output from the predict method to a dense format.
 
         Parameters
@@ -62,7 +62,7 @@ def sparse_to_dense(y_sparse: pd.Series, index: pd.Index) -> pd.Series[int]:
         return y_dense
 
     @staticmethod
-    def dense_to_sparse(y_dense: pd.Series) -> pd.Series[int]:
+    def dense_to_sparse(y_dense: pd.Series) -> pd.Series:
         """Convert the dense output from the transform method to a sparse format.
 
         Parameters