From 6495516e81c810362c10701c5c84a7f2fc2e25cc Mon Sep 17 00:00:00 2001
From: Kevin Schmidt <kevin3.schmidt@tu-dortmund.de>
Date: Thu, 18 Jul 2024 14:44:19 +0200
Subject: [PATCH] add torch to dependencies

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 1090f2f..0139550 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -33,6 +33,7 @@ requires-python = ">=3.10"
 dependencies = [
     "numpy",
     "astropy<=6.1.0",
+    "pytorch",
     "matplotlib",
     "ipython",
     "scipy",