[data.indian] Use right split name; separate split creation so it is reusable

src/ptbench/data/indian/datamodule.py

@@ -7,6 +7,15 @@ import importlib.resources
 from ..datamodule import CachingDataModule
 from ..shenzhen.datamodule import RawDataLoader
 from ..split import JSONDatabaseSplit
+from ..typing import DatabaseSplit
+
+
+def make_split(basename: str) -> DatabaseSplit:
+    """Returns a database split for the Indian database."""
+
+    return JSONDatabaseSplit(
+        importlib.resources.files(__name__.rsplit(".", 1)[0]).joinpath(basename)
+    )
 
 
 class DataModule(CachingDataModule):
@@ -42,10 +51,6 @@ class DataModule(CachingDataModule):
 
     def __init__(self, split_filename: str):
         super().__init__(
-            database_split=JSONDatabaseSplit(
-                importlib.resources.files(__name__.rsplit(".", 1)[0]).joinpath(
-                    split_filename
-                )
-            ),
+            database_split=make_split(split_filename),
             raw_data_loader=RawDataLoader(),
         )

src/ptbench/data/indian/default.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("default.json.bz2")
+datamodule = DataModule("default.json")

src/ptbench/data/indian/fold_0.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_0.json.bz2")
+datamodule = DataModule("fold_0.json")

src/ptbench/data/indian/fold_1.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_1.json.bz2")
+datamodule = DataModule("fold_1.json")

src/ptbench/data/indian/fold_2.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_2.json.bz2")
+datamodule = DataModule("fold_2.json")

src/ptbench/data/indian/fold_3.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_3.json.bz2")
+datamodule = DataModule("fold_3.json")

src/ptbench/data/indian/fold_4.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_4.json.bz2")
+datamodule = DataModule("fold_4.json")

src/ptbench/data/indian/fold_5.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_5.json.bz2")
+datamodule = DataModule("fold_5.json")

src/ptbench/data/indian/fold_6.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_6.json.bz2")
+datamodule = DataModule("fold_6.json")

src/ptbench/data/indian/fold_7.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_7.json.bz2")
+datamodule = DataModule("fold_7.json")

src/ptbench/data/indian/fold_8.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_8.json.bz2")
+datamodule = DataModule("fold_8.json")

src/ptbench/data/indian/fold_9.py

@@ -4,4 +4,4 @@
 
 from .datamodule import DataModule
 
-datamodule = DataModule("fold_9.json.bz2")
+datamodule = DataModule("fold_9.json")