From 1a47d1c99784d1247f41308d335b2087de13334c Mon Sep 17 00:00:00 2001
From: Alby M <1473644+Linux-cpp-lisp@users.noreply.github.com>
Date: Tue, 20 Apr 2021 12:03:13 -0600
Subject: Tests and miscellaneous (#4)

* linter

* Add tests

* Add restore test

* Type hints

* PyTorch dependency

* One fewer temp buffer
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'setup.py')

diff --git a/setup.py b/setup.py
index d998024..550c159 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ __version__ = '0.2'
 url = 'https://github.com/fadel/pytorch_ema'
 download_url = '{}/archive/{}.tar.gz'.format(url, __version__)
 
-install_requires = []
+install_requires = ["torch"]
 setup_requires = []
 tests_require = []
 
-- 
cgit v1.2.3