File tree 4 files changed +12
-12
lines changed
4 files changed +12
-12
lines changed Original file line number Diff line number Diff line change @@ -28,17 +28,17 @@ class MXNetProcessor(FrameworkProcessor):
28
28
29
29
def __init__ (
30
30
self ,
31
- framework_version , # New arg
31
+ framework_version ,
32
32
role ,
33
33
instance_count ,
34
34
instance_type ,
35
- py_version = "py3" , # New kwarg
35
+ py_version = "py3" ,
36
36
image_uri = None ,
37
37
command = None ,
38
38
volume_size_in_gb = 30 ,
39
39
volume_kms_key = None ,
40
40
output_kms_key = None ,
41
- code_location = None , # New arg
41
+ code_location = None ,
42
42
max_runtime_in_seconds = None ,
43
43
base_job_name = None ,
44
44
sagemaker_session = None ,
Original file line number Diff line number Diff line change @@ -28,17 +28,17 @@ class PyTorchProcessor(FrameworkProcessor):
28
28
29
29
def __init__ (
30
30
self ,
31
- framework_version , # New arg
31
+ framework_version ,
32
32
role ,
33
33
instance_count ,
34
34
instance_type ,
35
- py_version = "py3" , # New kwarg
35
+ py_version = "py3" ,
36
36
image_uri = None ,
37
37
command = None ,
38
38
volume_size_in_gb = 30 ,
39
39
volume_kms_key = None ,
40
40
output_kms_key = None ,
41
- code_location = None , # New arg
41
+ code_location = None ,
42
42
max_runtime_in_seconds = None ,
43
43
base_job_name = None ,
44
44
sagemaker_session = None ,
Original file line number Diff line number Diff line change @@ -28,17 +28,17 @@ class TensorFlowProcessor(FrameworkProcessor):
28
28
29
29
def __init__ (
30
30
self ,
31
- framework_version , # New arg
31
+ framework_version ,
32
32
role ,
33
33
instance_count ,
34
34
instance_type ,
35
- py_version = "py3" , # New kwarg
35
+ py_version = "py3" ,
36
36
image_uri = None ,
37
37
command = None ,
38
38
volume_size_in_gb = 30 ,
39
39
volume_kms_key = None ,
40
40
output_kms_key = None ,
41
- code_location = None , # New arg
41
+ code_location = None ,
42
42
max_runtime_in_seconds = None ,
43
43
base_job_name = None ,
44
44
sagemaker_session = None ,
Original file line number Diff line number Diff line change @@ -28,17 +28,17 @@ class XGBoostProcessor(FrameworkProcessor):
28
28
29
29
def __init__ (
30
30
self ,
31
- framework_version , # New arg
31
+ framework_version ,
32
32
role ,
33
33
instance_count ,
34
34
instance_type ,
35
- py_version = "py3" , # New kwarg
35
+ py_version = "py3" ,
36
36
image_uri = None ,
37
37
command = None ,
38
38
volume_size_in_gb = 30 ,
39
39
volume_kms_key = None ,
40
40
output_kms_key = None ,
41
- code_location = None , # New arg
41
+ code_location = None ,
42
42
max_runtime_in_seconds = None ,
43
43
base_job_name = None ,
44
44
sagemaker_session = None ,
You can’t perform that action at this time.
0 commit comments