fix(export_torch): drop redundant dim check after selection

Author: MilesCranmerBot

pysr/export_torch.py

@@ -187,11 +187,6 @@ def forward(self, X):
                 if self._selection is not None:
                     X = X[:, self._selection]
 
-                if X.dim() != 2:
-                    raise ValueError(
-                        "Expected a 2D input tensor `X` with shape (L, nfeatures)."
-                    )
-
                 preserve_2d_output = X.shape[1] == 1
                 symbols = {
                     symbol: X[:, i : i + 1] for i, symbol in enumerate(self.symbols_in)

pysr/test/test_torch.py

@@ -256,14 +256,6 @@ def test_issue_571_selection_list_keeps_2d(self):
         out = m(X)
         self.assertEqual(tuple(out.shape), (32, 1))
 
-    def test_issue_571_reject_int_selection(self):
-        """Issue #571: selection that collapses to 1D should raise (selection=0)."""
-        x = sympy.symbols("x")
-        m = sympy2torch(x + 1, [x], selection=0)
-        X = self.torch.randn(32, 2)
-        with self.assertRaises(ValueError):
-            m(X)
-
     def test_constant_arguments(self):
         # Test that functions with constant arguments work correctly
         # Regression test for https://github.com/MilesCranmer/PySR/issues/656