Helpful robot

Author: jessegrabowski

pymc/pytensorf.py

@@ -136,7 +136,7 @@ def _register_dataframe_backend(library_name: str):
 
         @_as_tensor_variable.register(library.Series)
         def series_to_tensor_variable(s: library.Series, *args, **kwargs) -> TensorVariable:
-            s = nw.from_native(s, allow_series=False)
+            s = nw.from_native(s, allow_series=True)
             if isinstance(s, nw.LazyFrame):
                 s = s.collect()
             return pt.as_tensor_variable(s.to_numpy(), *args, **kwargs)
@@ -149,6 +149,7 @@ def dataframe_to_tensor_variable(df: library.DataFrame, *args, **kwargs) -> Tens
             return pt.as_tensor_variable(df.to_numpy(), *args, **kwargs)
 
     except ImportError:
+        # Data backends are optional. Take no action if not installed.
         pass
 
 

tests/test_data.py

@@ -401,7 +401,7 @@ def test_implicit_coords_dataframe(self, seeded_test):
         assert "columns" in pmodel.coords
         assert pmodel.named_vars_to_dims == {"observations": ("rows", "columns")}
 
-    def test_implict_coords_polars_series(self):
+    def test_implicit_coords_polars_series(self):
         pl = pytest.importorskip("polars")
 
         ser_sales = pl.Series(