fix tox.ini

Author: benieric

src/sagemaker/modules/train/model_trainer.py

@@ -485,7 +485,7 @@ def model_post_init(self, __context: Any):
                         logger.info(f"contents: {contents}")
                         self.hyperparameters = yaml.safe_load(contents)
                         if not isinstance(self.hyperparameters, dict):
-                            raise ValueError(f"YAML contents must be a valid mapping")
+                            raise ValueError("YAML contents must be a valid mapping")
                         logger.info(f"hyperparameters: {self.hyperparameters}")
                         logger.debug("Hyperparameters loaded as YAML")
                     except (yaml.YAMLError, ValueError):

tox.ini

@@ -83,8 +83,8 @@ passenv =
 commands =
     python -c "import os; os.system('install-custom-pkgs --install-boto-wheels')"
     pip install 'apache-airflow==2.9.3' --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.9.3/constraints-3.8.txt"
-    pip install 'torch==2.0.1' -f 'https://download.pytorch.org/whl/torch_stable.html'
-    pip install 'torchvision==0.15.2' -f 'https://download.pytorch.org/whl/torch_stable.html'
+    pip install 'torch==2.0.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
+    pip install 'torchvision==0.15.2+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'dill>=0.3.8'
 
     pytest {posargs}