fix type import

rwth-i6 · Jun 7, 2024 · 59bcf41 · 59bcf41
1 parent 9787cf5
commit 59bcf41
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/datasets/huggingface.py b/datasets/huggingface.py
@@ -1,13 +1,17 @@
 """
 https://huggingface.co/docs/datasets/
 """
+from __future__ import annotations
 
-from typing import Optional, Any, Union
+from typing import Optional, Any, Union, TYPE_CHECKING
 from sisyphus import *
 from sisyphus.delayed_ops import DelayedBase
 
 from i6_core.util import instanciate_delayed
 
+if TYPE_CHECKING:
+    from datasets import Split
+
 
 class DownloadAndPrepareHuggingFaceDatasetJob(Job):
     """
@@ -26,8 +30,6 @@ class DownloadAndPrepareHuggingFaceDatasetJob(Job):
 
     __sis_hash_exclude__ = {"split": None, "token": None}
 
-    from datasets import Split
-
     def __init__(
         self,
         path: Union[str, DelayedBase],