yo

mosaicml · Aug 15, 2024 · be57e8b · be57e8b
1 parent d50ba0f
commit be57e8b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/streaming/__init__.py b/streaming/__init__.py
@@ -6,6 +6,7 @@
 import streaming.multimodal as multimodal
 import streaming.text as text
 import streaming.vision as vision
+from streaming._version import __version__  # noqa: F401
 from streaming.base import (CSVWriter, JSONWriter, LocalDataset, MDSWriter, Stream,
                             StreamingDataLoader, StreamingDataset, TSVWriter, XSVWriter)
 

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -9,6 +9,10 @@
 import pytest
 from moto import mock_aws
 
+from tests.common.utils import compressed_local_remote_dir  # pyright: ignore # noqa: F401
+from tests.common.utils import get_free_tcp_port  # pyright: ignore # noqa: F401
+from tests.common.utils import local_remote_dir  # pyright: ignore # noqa: F401
+
 MY_BUCKET = 'streaming-test-bucket'
 MY_PREFIX = 'train'
 GCS_URL = 'https://storage.googleapis.com'