update

fugue-project · Dec 1, 2023 · 18c421b · 18c421b
1 parent 9e8995b
commit 18c421b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/fugue_polars/test_transform.py b/tests/fugue_polars/test_transform.py
@@ -67,7 +67,7 @@ def tr3(dfs: Iterable[pl.DataFrame]) -> Iterator[pl.DataFrame]:
         assert fdf.as_array() == []
 
 
-def test_polars_on_engines(spark_session, fugue_dask_client, fugue_ray_session):
+def test_polars_on_engines(spark_session, dask_session, ray_session):
     def tr1(df: pl.DataFrame) -> pl.DataFrame:
         tdf = df.with_columns(pl.lit(1, pl.Int32()).alias("c"))
         return tdf
@@ -92,6 +92,6 @@ def test(engine):
 
     test(spark_session)
 
-    test(fugue_dask_client)
+    test(dask_session)
 
-    test(fugue_ray_session)
+    test(ray_session)