From 60e8141d1540f2b71b65975a5b32ed1498527caa Mon Sep 17 00:00:00 2001
From: Steven Rieder <steven@rieder.nl>
Date: Tue, 6 Aug 2024 13:04:15 +0200
Subject: [PATCH 1/2] add requirements.txt based on documentation

---
 requirements.txt | 6 ++++++
 1 file changed, 6 insertions(+)
 create mode 100644 requirements.txt

diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..5415c86
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+torch
+torchode
+numpy
+matplotlib
+natsort
+tqdm

From 95c2b3e3effd8a545d0c85f8670f587c46f64c5d Mon Sep 17 00:00:00 2001
From: Steven Rieder <steven@rieder.nl>
Date: Tue, 6 Aug 2024 13:08:19 +0200
Subject: [PATCH 2/2] Add scipy as a requirement, since MACE doesn't run
 without it.

---
 docs/src/index.rst | 1 +
 requirements.txt   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/docs/src/index.rst b/docs/src/index.rst
index d94ff4c..40b1638 100644
--- a/docs/src/index.rst
+++ b/docs/src/index.rst
@@ -57,6 +57,7 @@ The script run.py trains the model, as explained by `Maes et al. (2024) <https:/
     * matplotlib
     * natsort
     * tqdm
+    * scipy
 
 
 
diff --git a/requirements.txt b/requirements.txt
index 5415c86..9c04e9b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,4 @@ numpy
 matplotlib
 natsort
 tqdm
+scipy