fixed typing for variable names in fit

MilesCranmer · Aug 10, 2024 · 39cab90 · 39cab90
1 parent 1fe3fe3
commit 39cab90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/regressor_sequence.py b/pysr/regressor_sequence.py
@@ -104,7 +104,7 @@ def fit(
         self,
         X,
         weights=None,
-        variable_names: Optional[ArrayLike[str]] = None,
+        variable_names: Optional[List[str]] = None,
         complexity_of_variables: Optional[
             Union[int, float, List[Union[int, float]]]
         ] = None,