FIX: src/stimulus/data/{data_handlers.py,handlertorch.py} -> Change t…

…he docstring to match the right parameters
mathysgrapotte · Feb 18, 2025 · 0012c72 · 0012c72
1 parent eaaf40c
commit 0012c72
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/stimulus/data/data_handlers.py b/src/stimulus/data/data_handlers.py
@@ -293,7 +293,7 @@ def __init__(
         """Initialize the DatasetHandler with required config.
 
         Args:
-            config_path (str): Path to the dataset configuration file.
+            data_config (yaml_data.YamlSplitTransformDict): A YamlSplitTransformDict object holding the config.
             csv_path (str): Path to the CSV data file.
         """
         self.dataset_manager = DatasetManager(data_config)

diff --git a/src/stimulus/data/handlertorch.py b/src/stimulus/data/handlertorch.py
@@ -21,7 +21,7 @@ def __init__(
         """Initialize the TorchDataset.
 
         Args:
-            config_path: Path to the configuration file
+            data_config: A YamlSplitTransformDict holding the configuration.
             csv_path: Path to the CSV data file
             encoder_loader: Encoder loader instance
             split: Optional tuple containing split information