@@ -585,7 +585,6 @@ def model_code_key_prefix(code_location_key_prefix, model_name, image):
585
585
The location returned is a potential concatenation of 2 parts
586
586
1. code_location_key_prefix if it exists
587
587
2. model_name or a name derived from the image
588
-
589
588
Args:
590
589
code_location_key_prefix (str): the s3 key prefix from code_location
591
590
model_name (str): the name of the model
@@ -690,7 +689,8 @@ def validate_smdistributed(
690
689
instance_type (str): A string representing the type of training instance selected.
691
690
framework_name (str): A string representing the name of framework selected.
692
691
framework_version (str): A string representing the framework version selected.
693
- py_version (str): A string representing the python version selected. Ex: `py38, py39, py310, py311`
692
+ py_version (str): A string representing the python version selected.
693
+ Ex: `py38, py39, py310, py311`
694
694
distribution (dict): A dictionary with information to enable distributed training.
695
695
(Defaults to None if distributed training is not enabled.) For example:
696
696
@@ -762,7 +762,8 @@ def _validate_smdataparallel_args(
762
762
instance_type (str): A string representing the type of training instance selected. Ex: `ml.p3.16xlarge`
763
763
framework_name (str): A string representing the name of framework selected. Ex: `tensorflow`
764
764
framework_version (str): A string representing the framework version selected. Ex: `2.3.1`
765
- py_version (str): A string representing the python version selected. Ex: `py38, py39, py310, py311`
765
+ py_version (str): A string representing the python version selected.
766
+ Ex: `py38, py39, py310, py311`
766
767
distribution (dict): A dictionary with information to enable distributed training.
767
768
(Defaults to None if distributed training is not enabled.) Ex:
768
769
@@ -845,7 +846,8 @@ def validate_distribution(
845
846
instance_groups ([InstanceGroup]): A list contains instance groups used for training.
846
847
framework_name (str): A string representing the name of framework selected.
847
848
framework_version (str): A string representing the framework version selected.
848
- py_version (str): A string representing the python version selected. Ex: `py38, py39, py310, py311`
849
+ py_version (str): A string representing the python version selected.
850
+ Ex: `py38, py39, py310, py311`
849
851
image_uri (str): A string representing a Docker image URI.
850
852
kwargs(dict): Additional kwargs passed to this function
851
853
@@ -1008,7 +1010,8 @@ def validate_torch_distributed_distribution(
1008
1010
}
1009
1011
}
1010
1012
framework_version (str): A string representing the framework version selected.
1011
- py_version (str): A string representing the python version selected. Ex: `py38, py39, py310, py311`
1013
+ py_version (str): A string representing the python version selected.
1014
+ Ex: `py38, py39, py310, py311`
1012
1015
image_uri (str): A string representing a Docker image URI.
1013
1016
entry_point (str or PipelineVariable): The absolute or relative path to the local Python
1014
1017
source file that should be executed as the entry point to
@@ -1159,7 +1162,8 @@ def validate_version_or_image_args(framework_version, py_version, image_uri):
1159
1162
1160
1163
Args:
1161
1164
framework_version (str): The version of the framework.
1162
- py_version (str): A string representing the python version selected. Ex: `py38, py39, py310, py311`
1165
+ py_version (str): A string representing the python version selected.
1166
+ Ex: `py38, py39, py310, py311`
1163
1167
image_uri (str): The URI of the image.
1164
1168
1165
1169
Raises:
0 commit comments