Fix: requirements

Author: juliagusak

requirements.txt

@@ -3,4 +3,4 @@ scipy
 scikit-tensor-py3
 absl-py
 flopco-pytorch
-tensorly
+tensorly==0.4.5

setup.py

@@ -10,17 +10,20 @@
 
 def load_requirements(file_name):
     requirements = parse_requirements(file_name, session="test")
-    return [str(item.req) for item in requirements]
+    try:
+        return [str(item.req) for item in requirements]
+    except:
+        return [str(item.requirement) for item in requirements]
 
 
 setup(
     name="musco-pytorch",
-    version="1.0.4",
+    version="1.0.5",
     description="MUSCO: Multi-Stage COmpression of Neural Networks",
     author="Julia Gusak, Maksym Kholiavchenko, Evgeny Ponomarev, Larisa Markeeva, Andrzej Cichocki, Ivan Oseledets",
     author_email="julgusak@gmail.com",
     url="https://github.com/musco-ai/musco-pytorch",
-    download_url="https://github.com/musco-ai/musco-pytorch/archive/1.0.4.tar.gz",
+    download_url="https://github.com/musco-ai/musco-pytorch/archive/1.0.5.tar.gz",
     license="Apache-2.0",
     packages=find_packages(),
     install_requires=load_requirements("requirements.txt")