Fix Tests for PT 2.8 Training SM (aws#5526)

* Fix Package Regression for langcodes

* revert toml

---------

Co-authored-by: DevakiBolleneni <devakib@amazon.com>

Author: DevakiBolleneni

pytorch/training/docker/2.8/py3/Dockerfile.cpu

@@ -299,7 +299,8 @@ RUN pip install --no-cache-dir -U \
     scikit-learn \
     seaborn \
     shap \
-    cloudpickle 
+    cloudpickle \
+    "langcodes<=3.5.0"
 
 # Copy workaround script for incorrect hostname
 COPY changehostname.c /

pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu

@@ -267,7 +267,8 @@ RUN pip install --no-cache-dir -U \
     shap \
     scikit-learn \
     seaborn \
-    cloudpickle 
+    cloudpickle \
+    "langcodes<=3.5.0"
 
 COPY setup_oss_compliance.sh setup_oss_compliance.sh
 RUN bash setup_oss_compliance.sh ${PYTHON} && rm setup_oss_compliance.sh