Add typing hints

theochem · Oct 6, 2024 · fafc517 · fafc517
1 parent a7de226
commit fafc517
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/selector/methods/base.py b/selector/methods/base.py
@@ -25,6 +25,7 @@
 
 import warnings
 from abc import ABC, abstractmethod
+from typing import List, Iterable, Union
 
 import numpy as np
 
@@ -40,7 +41,7 @@ def select(
         size: int,
         labels: np.ndarray = None,
         proportional_selection: bool = True,
-    ) -> list:
+    ) -> Union[List, Iterable]:
         """Return indices representing subset of sample points.
 
         Parameters