Use Union to support Python<3.10

allenai · Jan 15, 2025 · 2b354a4 · 2b354a4
1 parent 878cc8c
commit 2b354a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/olmo/util.py b/olmo/util.py
@@ -715,7 +715,7 @@ def _http_get_bytes_range(scheme: str, host_name: str, path: str, bytes_start: i
 
 
 def save_hf_dataset_to_disk(
-    dataset: datasets.DatasetDict | datasets.Dataset,
+    dataset: Union[datasets.DatasetDict, datasets.Dataset],
     hf_path: str,
     name: Optional[str],
     split: str,