test: fix typing error

Author: MilesCranmer

pysr/sr.py

@@ -110,7 +110,7 @@ def _process_constraints(
 
             # Apply arity-specific validation for existing constraints
             if isinstance(constraints[op], tuple):
-                constraint_tuple: tuple[int, ...] = constraints[op]
+                constraint_tuple = cast(Tuple[int, ...], constraints[op])
                 # Validate that constraint tuple length matches operator arity
                 if len(constraint_tuple) != arity:
                     raise ValueError(

pysr/test/test_main.py

@@ -1294,7 +1294,7 @@ def test_deprecated_functions(self):
     def test_power_law_warning(self):
         """Ensure that a warning is given for a power law operator."""
         with self.assertWarns(UserWarning):
-            _process_constraints(["^"], [], {})
+            _process_constraints({2: ["^"]}, {})
 
     def test_size_warning(self):
         """Ensure that a warning is given for a large input size."""
@@ -1420,7 +1420,7 @@ def test_suggest_keywords(self):
 
         # Farther matches (this might need to be changed)
         with self.assertRaises(TypeError) as cm:
-            PySRRegressor(operators=["+", "-"])
+            PySRRegressor(nary_operators=["+", "-"])
 
         self.assertIn("`unary_operators`, `binary_operators`", str(cm.exception))