From 767206f9053bf7b597da785be07c794303389683 Mon Sep 17 00:00:00 2001
From: Alexander Engelsberger <aengelsb@hs-mittweida.de>
Date: Tue, 11 May 2021 13:41:09 +0200
Subject: [PATCH] Define minimum prototorch version in setup

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index faeddcb..2a61840 100644
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@ DOWNLOAD_URL = "https://github.com/si-cim/prototorch_models.git"
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-INSTALL_REQUIRES = ["prototorch", "pytorch_lightning", "torchmetrics"]
+INSTALL_REQUIRES = ["prototorch>=0.4.1", "pytorch_lightning", "torchmetrics"]
 DEV = ["bumpversion"]
 EXAMPLES = ["matplotlib", "scikit-learn"]
 TESTS = ["codecov", "pytest"]