Convert int columns in spaceship data to ints (#300)

sodascience · Aug 8, 2024 · 6ddfbf8 · 6ddfbf8
1 parent 7bd6a0a
commit 6ddfbf8
Show file tree

Hide file tree

Showing 2 changed files with 8,696 additions and 8,694 deletions.
diff --git a/metasyn/demo/dataset.py b/metasyn/demo/dataset.py
@@ -126,7 +126,9 @@ def demo_dataframe(name: str = "titanic") -> pl.DataFrame:
             "Destination": pl.Categorical,
             "Transported": pl.Categorical,
         }
-        return pl.read_csv(file_path, schema_overrides=data_types, try_parse_dates=True)
+        return pl.read_csv(
+            file_path, schema_overrides=data_types, try_parse_dates=True
+        )
     if name == "titanic":
         # our edited titanic data
         data_types = {"Sex": pl.Categorical, "Embarked": pl.Categorical}