Upload load.py with huggingface_hub

load.py

@@ -7,7 +7,9 @@ from .operator import StreamSource
 
 
 def load_dataset(source: Union[StreamSource, str]) -> DatasetDict:
-    assert isinstance(source, (StreamSource, str)), "source must be a StreamSource or a string"
+    assert isinstance(
+        source, (StreamSource, str)
+    ), "source must be a StreamSource or a string"
     if isinstance(source, str):
         source, _ = fetch_artifact(source)
     return source().to_dataset()