Skip to content

Commit e80127d

Browse files
committed
v1.6.1: enhance code style
1 parent 5448450 commit e80127d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
with open("README.md", "r", encoding="utf-8") as fh:
66
long_description = fh.read()
77

8-
with open("requirements.txt", "r", encoding="utf-8") as fh:
9-
requirements = [line.strip() for line in fh if line.strip() and not line.startswith("#")]
8+
# Dependencies
9+
requirements = ["torch>=2.0", "numpy"]
1010

1111
setuptools.setup(
1212
name="torch-pruning",

0 commit comments

Comments
 (0)