From d0be718d443f8d676640b3aa75a7a154edad5dce Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Fri, 29 May 2020 15:33:56 +0200
Subject: [PATCH] update torch requirements

need it for the mixtureSameFamily
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 2e69c81..6633866 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@ setup(
     zip_safe=True,
     python_requires=">=3.6",
     install_requires = [
-        'torch>=1.3.0',
+        'torch>=1.5.0',
         'holidays',
         'numpy',
         'pandas',