cicd: fixing dependencies

lucas-nelson-uiuc · Nov 28, 2024 · b1b10a5 · b1b10a5
1 parent fc77e92
commit b1b10a5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_selector.py b/tests/test_selector.py
@@ -4,6 +4,7 @@
 from pyspark.sql import SparkSession, types as T
 
 from tidy_tools.core import selector as cs
+from tidy_tools.tidy import TidyDataFrame
 
 
 @pytest.fixture
@@ -14,7 +15,7 @@ def spark_fixture():
 
 @pytest.fixture
 def sample_data(spark_fixture):
-    yield spark_fixture.createDataFrame(
+    data = spark_fixture.createDataFrame(
         [
             {
                 "name": "Homer",
@@ -55,6 +56,7 @@ def sample_data(spark_fixture):
             ]
         ),
     )
+    yield TidyDataFrame(data)
 
 
 class TestColumnSelector: