Fix up other arguments in test

test/test_jax.py

@@ -38,9 +38,7 @@ class TestJAX(unittest.TestCase):
         model = PySRRegressor(
             equation_file="equation_file.csv",
             output_jax_format=True,
-            variables_names="x1 x2 x3".split(" "),
-            multioutput=False,
-            nout=1,
+            variable_names="x1 x2 x3".split(" "),
             selection=[1, 2, 3],
         )
 

test/test_torch.py

@@ -37,13 +37,11 @@ class TestTorch(unittest.TestCase):
         model = PySRRegressor(
             model_selection="accuracy",
             equation_file="equation_file.csv",
-            variables_names="x1 x2 x3".split(" "),
+            variable_names="x1 x2 x3".split(" "),
             extra_sympy_mappings={},
             output_torch_format=True,
-            multioutput=False,
-            nout=1,
-            selection=[1, 2, 3],
         )
+        model.selection = [1, 2, 3]
         model.n_features = 2  # TODO: Why is this 2 and not 3?
         model.using_pandas = False
         model.refresh()
@@ -91,14 +89,12 @@ class TestTorch(unittest.TestCase):
         model = PySRRegressor(
             model_selection="accuracy",
             equation_file="equation_file_custom_operator.csv",
-            variables_names="x1 x2 x3".split(" "),
+            variable_names="x1 x2 x3".split(" "),
             extra_sympy_mappings={"mycustomoperator": sympy.sin},
             extra_torch_mappings={"mycustomoperator": torch.sin},
             output_torch_format=True,
-            multioutput=False,
-            nout=1,
-            selection=[0, 1, 2],
         )
+        model.selection = [0, 1, 2]
         model.n_features = 3
         model.using_pandas = False
         model.refresh()