From a24bbde1ffb5d536ac1962216f2c325b7233a92a Mon Sep 17 00:00:00 2001
From: Wessel Bruinsma <wessel.p.bruinsma@gmail.com>
Date: Tue, 6 Aug 2024 16:30:38 +0200
Subject: [PATCH] Fix versions

---
 pyproject.toml | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 0c4c99c..d84e34c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,9 +32,9 @@ dynamic = ["version"]
 
 requires-python = ">=3.10"
 dependencies = [
-    "numpy",
-    "torch",
-    "einops",
+    "numpy==1.26.2",
+    "torch==2.2.0",
+    "einops==0.6.1",
     "timm==0.6.13",
     "huggingface-hub",
 ]
@@ -50,7 +50,6 @@ dev = [
     "ruff==0.4.1",
     "pre-commit",
     "jupyter-book",
-    "torch==2.2.0",
 ]
 
 [project.urls]