un-skipping gather test

pola-rs · Jul 7, 2024 · 73e12ee · 73e12ee
1 parent 7980f50
commit 73e12ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/__tests__/expr.test.ts b/__tests__/expr.test.ts
@@ -917,7 +917,7 @@ describe("expr", () => {
     const actual = df.select(col("a").tail(3).as("tail3"));
     expect(actual).toFrameEqual(expected);
   });
-  test.skip("take", () => {
+  test("take", () => {
     const df = pl.DataFrame({ a: [1, 2, 2, 3, 3, 8, null, 1] });
     const expected = pl.DataFrame({
       "take:array": [1, 2, 3, 8],

diff --git a/src/lazy/dsl.rs b/src/lazy/dsl.rs
@@ -330,7 +330,7 @@ impl JsExpr {
     }
     #[napi(catch_unwind)]
     pub fn gather(&self, idx: &JsExpr) -> JsExpr {
-        self.clone().inner.gather(idx.inner.clone()).into()
+        self.clone().inner.gather(idx.inner.clone().cast(DataType::Int64)).into()
     }
     #[napi(catch_unwind)]
     pub fn sort_by(&self, by: Vec<&JsExpr>, reverse: Vec<bool>) -> JsExpr {