Skip to content

enable_network_isolation documented but not implemented in remote jobs. #4425

Description

@ArtemioPadilla

Describe the bug
A clear and concise description of what the bug is.
The parameter enable_network_isolation is documented in the RemoteExector, yet it is not a usable parameter for either the decorator remote or the class.

To reproduce

Just try to use enable_network_isolation set as True and run anything.

Expected behavior
SageMaker job ran with enable_network_isolation configured as True.

Screenshots or logs
---> 10 @Remote(instance_type="ml.c5.xlarge",
11 sagemaker_session=session,
12 job_name_prefix="test-job",
23
24 enable_network_isolation=True
25 )
26 def matrix_multiply(a, b):
27 return np.matmul(a, b)
29 a = np.array([[1, 0],
30 [0, 1]])

TypeError: remote() got an unexpected keyword argument 'enable_network_isolation'

System information
SageMaker Jupyter Lab from New SageMaker Studio- SageMaker Python SDK version: 2.198.0

  • Custom Docker image (Y/N): N

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions