added torchvision version

CrayLabs · Aug 20, 2024 · ae32e53 · ae32e53
1 parent e7c1882
commit ae32e53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/smartsim/_core/_install/buildenv.py b/smartsim/_core/_install/buildenv.py
@@ -217,6 +217,7 @@ def ml_extras_required(self) -> t.Dict[str, t.List[str]]:
             "skl2onnx": "1.16.0",
             "onnxmltools": "1.12.0",
             "scikit-learn": "1.3.2",
+            "torchvision": "0.15.2",
         }
 
         # remove torch-related fields as they are subject to change