Fix: setup.py

flopco/flopco.py

@@ -213,4 +213,4 @@ def get_stats(self, shapes = True, flops = False, macs = False, params = False):
                 m._forward_pre_hooks.clear()
                 m._forward_hooks.clear()
 
-        torch.cuda.empty_cache()
+        torch.cuda.empty_cache()

setup.py

@@ -11,17 +11,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="flopco-pytorch",
-    version="v0.1.3",
+    version="v0.1.4",
     description="FLOPs and other statistics COunter for Pytorch neural networks",
     author="Julia Gusak",
     author_email="[email protected]",
     url="https://github.com/juliagusak/flopco-pytorch",
-    download_url="https://github.com/juliagusak/flopco-pytorch/archive/v0.1.3.tar.gz",
+    download_url="https://github.com/juliagusak/flopco-pytorch/archive/v0.1.4.tar.gz",
     keywords = ['pytorch', 'flops', 'macs', 'neural-networks', 'cnn'],
     license="MIT",
     packages=find_packages(),