From 39cab903966c2dd853d26677208b2d2636ed8699 Mon Sep 17 00:00:00 2001 From: wenbang24 Date: Sat, 10 Aug 2024 12:18:32 +1000 Subject: [PATCH] fixed typing for variable names in fit --- pysr/regressor_sequence.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pysr/regressor_sequence.py b/pysr/regressor_sequence.py index efd42bd9..b5f4737a 100644 --- 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,