sort version string

Rui Wang Napieralski · Rui Wang Napieralski · commit e6873d17d1f3 · 2021-03-24T10:10:03.000-07:00
diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py
@@ -59,8 +59,8 @@
     "local_gpu",
 )
 SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {
-    "tensorflow": ["2.3.1", "2.3.2", "2.4.1", "2.4", "2.3"],
-    "pytorch": ["1.6.0", "1.7.1", "1.8.0", "1.6", "1.7", "1.8"],
+    "tensorflow": ["2.3", "2.3.1", "2.3.2", "2.4", "2.4.1"],
+    "pytorch": ["1.6", "1.6.0", "1.7", "1.7.1", "1.8", "1.8.0"],
 }
 SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]
 

Original file line number	Diff line number	Diff line change
`@@ -59,8 +59,8 @@`
`59`	`59`	`"local_gpu",`
`60`	`60`	`)`
`61`	`61`	`SM_DATAPARALLEL_SUPPORTED_FRAMEWORK_VERSIONS = {`
`62`		`- "tensorflow": ["2.3.1", "2.3.2", "2.4.1", "2.4", "2.3"],`
`63`		`- "pytorch": ["1.6.0", "1.7.1", "1.8.0", "1.6", "1.7", "1.8"],`
	`62`	`+ "tensorflow": ["2.3", "2.3.1", "2.3.2", "2.4", "2.4.1"],`
	`63`	`+ "pytorch": ["1.6", "1.6.0", "1.7", "1.7.1", "1.8", "1.8.0"],`
`64`	`64`	`}`
`65`	`65`	`SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]`
`66`	`66`