From be57e8b8f88e89ff375ac536fbc7f357f15255e0 Mon Sep 17 00:00:00 2001
From: Saaketh <narayan.saaketh@gmail.com>
Date: Wed, 14 Aug 2024 21:42:59 -0700
Subject: [PATCH] yo

---
 streaming/__init__.py | 1 +
 tests/conftest.py     | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/streaming/__init__.py b/streaming/__init__.py
index e7b98dcca..b8f3e2541 100644
--- 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
index adf131628..191b50af9 100644
--- 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'