Skip to content

Commit 094e476

Browse files
committed
fix: pytorch compiler unit tests
1 parent eb3ba1b commit 094e476

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/sagemaker/training_compiler/test_pytorch_compiler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from sagemaker.pytorch import PyTorch, TrainingCompilerConfig
2525
from sagemaker.pytorch.model import PyTorchModel
2626
from sagemaker.instance_group import InstanceGroup
27+
from sagemaker.session_settings import SessionSettings
2728

2829
from tests.unit.sagemaker.training_compiler import EC2_GPU_INSTANCE_CLASSES
2930

@@ -71,6 +72,7 @@ def fixture_sagemaker_session():
7172
local_mode=False,
7273
s3_resource=None,
7374
s3_client=None,
75+
settings=SessionSettings(),
7476
)
7577

7678
describe = {"ModelArtifacts": {"S3ModelArtifacts": "s3://m/m.tar.gz"}}

0 commit comments

Comments
 (0)