From c050d652217704fd059e5d2e63aeb6b796f980e3 Mon Sep 17 00:00:00 2001
From: Mayank Mishra <mayank31398@gmail.com>
Date: Thu, 16 Apr 2026 19:50:06 -0700
Subject: [PATCH] add w

Signed-off-by: Mayank Mishra <mayank31398@gmail.com>
---
 pyproject.toml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 77813e63a..1ad37026a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ Repository = "https://github.com/open-lm-engine/lm-engine"
 
 [project.optional-dependencies]
 cuda = [
-    "torch>=2.10.0",
+    "torch==2.10.0",
 ]
 dev = [
     "pre-commit>=4.4.0",
@@ -46,12 +46,12 @@ tpu = [
 ]
 data = [
     "ray[default]>=2.52.1",
-    "torch>=2.10.0",
+    "torch==2.10.0",
     "zstandard>=0.25.0",
     "aiobotocore>=3.0.0",
 ]
 torch = [
-    "torch>=2.10.0",
+    "torch==2.10.0",
 ]
 
 [tool.hatch.build.targets.wheel]