Specifying return type for TransitionSolver.predict()

washingtonpost · Mar 19, 2024 · b5a2e07 · b5a2e07
1 parent 71f29c9
commit b5a2e07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/elexsolver/TransitionSolver.py b/src/elexsolver/TransitionSolver.py
@@ -39,7 +39,7 @@ def fit(self, X: np.ndarray, Y: np.ndarray, sample_weight: np.ndarray | None = N
         """
         raise NotImplementedError
 
-    def predict(self, X: np.ndarray):
+    def predict(self, X: np.ndarray) -> np.ndarray:
         """
         Parameters
         ----------