test fix

pagezyhf · pagezyhf · commit 73c723f1f936 · 2025-10-27T10:17:49.000+01:00
diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py
@@ -156,6 +156,9 @@
     "2.3.1",
     "2.4.1",
     "2.5.1",
+    "2.6.0", 
+    "2.7.1",
+    "2.8.0",
 ]
 
 TRAINIUM_SUPPORTED_DISTRIBUTION_STRATEGIES = ["torch_distributed"]

Original file line number	Diff line number	Diff line change
`@@ -156,6 +156,9 @@`
`156`	`156`	`"2.3.1",`
`157`	`157`	`"2.4.1",`
`158`	`158`	`"2.5.1",`
	`159`	`+ "2.6.0",`
	`160`	`+ "2.7.1",`
	`161`	`+ "2.8.0",`
`159`	`162`	`]`
`160`	`163`
`161`	`164`	`TRAINIUM_SUPPORTED_DISTRIBUTION_STRATEGIES = ["torch_distributed"]`