Create split file directory and write with indent

Author: samedii

datastream/dataset.py

@@ -206,17 +206,12 @@ def split(
         ...     )))
         ...     .split(
         ...         key_column='index',
-        ...         proportions=dict(
-        ...             train=0.7,
-        ...             early_stopping=0.1,
-        ...             public_test=0.1,
-        ...             private_test=0.1,
-        ...         ),
+        ...         proportions=dict(train=0.8, test=0.2),
         ...         filepath=split_file,
         ...     )
         ... )
         >>> len(split_datasets['train'])
-        70
+        80
         >>> split_file.unlink()  # clean up after doctest
         '''
         if filepath is not None:

datastream/tools/split_dataframes.py

@@ -72,7 +72,8 @@ def split_dataframes(
         split[last_split_name] += unassigned(dataframe[key_column], split)
 
         if filepath is not None:
-            filepath.write_text(json.dumps(split))
+            filepath.parent.mkdir(parents=True, exist_ok=True)
+            filepath.write_text(json.dumps(split, indent=4))
 
     return {
         split_name: (