fix(tests): resolve e2e test import errors after adding __init__ files

- Fix relative imports in e2e test files to use proper package paths
- Add type ignore comment for spy function in dcp optimized tests

Resolves import errors introduced when adding __init__.py files to make
test directories Python packages (for READER_TYPE_STRING_TO_CLASS)
changes).

Author: jet-tong

s3torchconnector/tst/e2e/dcp/test_e2e_s3_file_system.py

@@ -29,7 +29,7 @@
 from typing import Optional
 
 from s3torchconnector.dcp.s3_prefix_strategy import RoundRobinPrefixStrategy
-from test_common import _list_folders_in_bucket
+from ..test_common import _list_folders_in_bucket
 
 
 def generate_random_port():

s3torchconnector/tst/e2e/test_distributed_training.py

@@ -20,7 +20,7 @@
     from .conftest import BucketPrefixFixture, BucketPrefixData
 
 
-from test_common import _get_fork_methods, _read_data, _set_start_method
+from .test_common import _get_fork_methods, _read_data, _set_start_method
 
 
 start_methods = _get_fork_methods()

s3torchconnector/tst/e2e/test_e2e_s3_lightning_checkpoint.py

@@ -19,8 +19,8 @@
 from s3torchconnector.lightning import S3LightningCheckpoint
 from s3torchconnectorclient import S3Exception, __version__
 
-from models.net import Net
-from models.lightning_transformer import LightningTransformer, L
+from .models.net import Net
+from .models.lightning_transformer import LightningTransformer, L
 
 
 LIGHTNING_ACCELERATOR = "cpu"

s3torchconnector/tst/e2e/test_e2e_s3checkpoint.py

@@ -5,7 +5,7 @@
 import pytest
 
 from s3torchconnector import S3Checkpoint
-from models.net import Net
+from .models.net import Net
 
 
 @pytest.mark.parametrize(

s3torchconnector/tst/e2e/test_mountpoint_client_parallel_access.py

@@ -5,8 +5,8 @@
 from s3torchconnector._s3client import S3Client
 from s3torchconnectorclient._mountpoint_s3_client import MountpointS3Client
 
-from test_common import _get_fork_methods
-from conftest import getenv
+from .test_common import _get_fork_methods
+from .conftest import getenv
 
 
 NATIVE_S3_CLIENT = None

s3torchconnector/tst/e2e/test_multiprocess_dataloading.py

@@ -22,7 +22,7 @@
 if TYPE_CHECKING:
     from .conftest import BucketPrefixFixture
 
-from test_common import _get_fork_methods, _read_data, _set_start_method
+from .test_common import _get_fork_methods, _read_data, _set_start_method
 
 
 start_methods = _get_fork_methods()

s3torchconnector/tst/unit/test_s3reader_dcp_optimized.py

@@ -524,7 +524,7 @@ def spy_get_stream(start=None, end=None):
             TEST_KEY,
             ranges,
             create_object_info_getter(test_data),
-            spy_get_stream,
+            spy_get_stream,  # type: ignore
             max_gap_size=10,
         )