From 9ede5ea1593b7d53572f6ebd6f951932ff541354 Mon Sep 17 00:00:00 2001
From: Gaurav Saini <147703805+gauravsaini04@users.noreply.github.com>
Date: Thu, 8 Aug 2024 21:54:55 +0530
Subject: [PATCH] [universal] - reduce size - install prev. rel. of torch
 (#1147)

---
 .../local-features/machine-learning-packages/install.sh         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/universal/.devcontainer/local-features/machine-learning-packages/install.sh b/src/universal/.devcontainer/local-features/machine-learning-packages/install.sh
index c16153a2a..917a04d28 100644
--- a/src/universal/.devcontainer/local-features/machine-learning-packages/install.sh
+++ b/src/universal/.devcontainer/local-features/machine-learning-packages/install.sh
@@ -44,7 +44,7 @@ if [[ "$(python --version)" != "" ]] && [[ "$(pip --version)" != "" ]]; then
     install_python_package "matplotlib"
     install_python_package "seaborn"
     install_python_package "scikit-learn"
-    install_python_package "torch" "-f https://download.pytorch.org/whl/cpu/torch_stable.html"
+    install_python_package "torch==2.3.1" "-f https://download.pytorch.org/whl/cpu/torch_stable.html"
     install_python_package "requests"
     install_python_package "plotly"
 else