Skip to content

Commit ce7ec82

Browse files
committed
fix: docker_compose_version test
Use FileSystemDataSource instead of S3DataSource to avoid triggering Session() creation in model_post_init. The test only exercises _get_compose_cmd_prefix() and needs no AWS access.
1 parent 40b36ef commit ce7ec82

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

‎sagemaker-train/tests/integ/train/test_docker_compose_version_detection.py‎

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from sagemaker.core.modules.local_core.local_container import (
3131
_LocalContainer as CoreModulesLocalContainer,
3232
)
33-
from sagemaker.core.shapes import Channel, DataSource, S3DataSource
33+
from sagemaker.core.shapes import Channel, DataSource, FileSystemDataSource
3434
from sagemaker.train.local.local_container import (
3535
_LocalContainer as TrainLocalContainer,
3636
)
@@ -63,10 +63,11 @@ def _get_installed_compose_major_version():
6363
def _make_basic_channel():
6464
"""Create a minimal Channel for constructing _LocalContainer instances."""
6565
data_source = DataSource(
66-
s3_data_source=S3DataSource(
67-
s3_uri="s3://bucket/data",
68-
s3_data_type="S3Prefix",
69-
s3_data_distribution_type="FullyReplicated",
66+
file_system_data_source=FileSystemDataSource(
67+
file_system_id="fs-0123456789abcdef0",
68+
file_system_access_mode="ro",
69+
file_system_type="EFS",
70+
directory_path="/tmp/sagemaker-integ-compose-data",
7071
)
7172
)
7273
return Channel(channel_name="training", data_source=data_source)

0 commit comments

Comments
 (0)