fix remote dataset config string

LoicGrobol · Apr 16, 2024 · 0326a4b · 0326a4b
1 parent 395c523
commit 0326a4b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -17,7 +17,7 @@ def raw_text_path(test_data_dir: pathlib.Path) -> pathlib.Path:
 
 @pytest.fixture(scope="session")
 def remote_raw_text() -> str:
-    return "lgrobol/openminuscule:text:train"
+    return "lgrobol/openminuscule:default:train"
 
 @pytest.fixture(scope="session")
 def translation_dataset_path(test_data_dir: pathlib.Path) -> pathlib.Path:

diff --git a/zeldarose/train_transformer.py b/zeldarose/train_transformer.py
@@ -323,7 +323,7 @@ def on_train_epoch_end(self, trainer: pl.Trainer, pl_module: TrainingModule):
     help=(
         "A raw corpus for validation."
         " Either as a path or as a `handle:config:split` identifier for 🤗 hub."
-        " (handle can be a url), e.g. `lgrobol/openminuscule:text:train`"
+        " (handle can be a url), e.g. `lgrobol/openminuscule:default:train`"
     ),
 )
 @click.option("--verbose", is_flag=True, help="More detailed logs")