We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eb3ba1b commit 094e476Copy full SHA for 094e476
tests/unit/sagemaker/training_compiler/test_pytorch_compiler.py
@@ -24,6 +24,7 @@
24
from sagemaker.pytorch import PyTorch, TrainingCompilerConfig
25
from sagemaker.pytorch.model import PyTorchModel
26
from sagemaker.instance_group import InstanceGroup
27
+from sagemaker.session_settings import SessionSettings
28
29
from tests.unit.sagemaker.training_compiler import EC2_GPU_INSTANCE_CLASSES
30
@@ -71,6 +72,7 @@ def fixture_sagemaker_session():
71
72
local_mode=False,
73
s3_resource=None,
74
s3_client=None,
75
+ settings=SessionSettings(),
76
)
77
78
describe = {"ModelArtifacts": {"S3ModelArtifacts": "s3://m/m.tar.gz"}}
0 commit comments