fix: update parquet export tests to match new schema

- Change 'goal' to 'instruction' in column assertions
- Change 'image_path' to 'screenshot_path' to match schema

Author: abrichr

tests/test_parquet_export.py

@@ -75,13 +75,13 @@ def test_to_parquet_basic(self, sample_episodes, tmp_path):
         # Check expected columns exist
         expected_columns = [
             "episode_id",
-            "goal",
+            "instruction",  # Changed from "goal" to match new schema
             "step_index",
             "timestamp",
             "action_type",
             "x",
             "y",
-            "image_path",
+            "screenshot_path",  # Changed from "image_path" to match new schema
         ]
         for col in expected_columns:
             assert col in table.column_names, f"Missing column: {col}"
@@ -108,7 +108,7 @@ def test_to_parquet_with_summary(self, sample_episodes, tmp_path):
         # Check summary columns
         assert "episode_id" in summary_table.column_names
         assert "step_count" in summary_table.column_names
-        assert "goal" in summary_table.column_names
+        assert "instruction" in summary_table.column_names  # Changed from "goal"
 
     def test_from_parquet_roundtrip(self, sample_episodes, tmp_path):
         """Test Parquet roundtrip (lossy but reconstructable)."""