From 1e642109dc7bb668c1e80f53ef80803d4ff11701 Mon Sep 17 00:00:00 2001
From: Charles Coulombe <ccoulombe@users.noreply.github.com>
Date: Tue, 23 Jan 2024 16:26:07 -0500
Subject: [PATCH] Added install requirements to setup (#488)

* Added install requirements to setup

* Update setup.py

Co-authored-by: Aarni Koskela <akx@iki.fi>

---------

Co-authored-by: Aarni Koskela <akx@iki.fi>
---
 setup.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/setup.py b/setup.py
index a71331dcf..b55678bf4 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,8 @@ def read(fname):
     install_requires=['scipy'],
     packages=find_packages(),
     package_data={"": libs},
+    install_requires=['torch', 'numpy', 'scipy'],
+    extras_require={'benchmark': ['pandas', 'matplotlib']},
     long_description=read("README.md"),
     long_description_content_type="text/markdown",
     classifiers=[