minor

kaizhang · Feb 28, 2024 · 9809ece · 9809ece
1 parent 9aab385
commit 9809ece
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/pyanndata/Cargo.toml b/pyanndata/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pyanndata"
-version = "0.3.0"
+version = "0.3.1"
 edition = "2021"
 rust-version = "1.70"
 authors = ["Kai Zhang <kai@kzhang.org>"]

diff --git a/pyanndata/src/container.rs b/pyanndata/src/container.rs
@@ -153,7 +153,8 @@ impl PyDataFrameElem {
         self.0.get(subscript)
     }
 
-    fn __setitem__(&self, key: &str, data: PySeries) -> Result<()> {
+    fn __setitem__(&self, key: &str, data: &PyAny) -> Result<()> {
+        let data: PySeries = data.py().import("polars")?.call_method1("Series", (data, ))?.extract()?;
         self.0.set(key, data.into())
     }
 

diff --git a/python/setup.py b/python/setup.py
@@ -17,6 +17,7 @@
         "anndata>=0.8",
         "numpy>=1.16",
         "pandas",
+        "pyarrow",
         "scipy>=1.4",
         "polars>=0.14",
     ],

diff --git a/python/tests/test_base.py b/python/tests/test_base.py
@@ -103,6 +103,7 @@ def test_creation(tmp_path):
     adata.uns['df'] = pd.DataFrame([[1, 2, 3], [4, 5, 6]], columns=["a", "b", "c"])
 
     adata.obs = pl.DataFrame({"a": ['1', '2'], "b": [3, 4]})
+    adata.obs['c'] = np.array([1, 2])
     obs_names = list(adata.obs['a'])
     adata.obs_names = obs_names
     adata.obs = pl.DataFrame()