Skip to content

Commit 7b5eb65

Browse files
committed
refactor: setup
1 parent 4dacfb0 commit 7b5eb65

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

setup.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,20 +13,22 @@ def read_file(filename: str) -> str:
1313
def read_version() -> str:
1414
regexp = re.compile(r"^__VERSION__\W*=\W*'([\d.abrc]+)'")
1515

16-
with open(os.path.join(os.path.dirname(__file__), 'pytorch_optimizer', '__init__.py')) as f:
16+
with open(os.path.join(os.path.dirname(__file__), 'pytorch_optimizer', 'version.py')) as f:
1717
for line in f:
1818
match = regexp.match(line)
1919
if match is not None:
2020
return match.group(1)
2121

22-
raise RuntimeError('Cannot find version in pytorch_optimizer/__init__.py')
22+
raise RuntimeError('Cannot find version in pytorch_optimizer/version.py')
2323

2424

25-
INSTALL_REQUIRES: List[str] = ['torch>=1.5.0']
25+
INSTALL_REQUIRES: List[str] = ['torch>=1.4.0']
2626

2727

2828
CLASSIFIERS: List[str] = [
2929
'License :: OSI Approved :: Apache Software License',
30+
'Development Status :: 3 - Alpha',
31+
'Intended Audience :: Education',
3032
'Intended Audience :: Developers',
3133
'Intended Audience :: Science/Research',
3234
'Programming Language :: Python :: 3',
@@ -35,7 +37,11 @@ def read_version() -> str:
3537
'Programming Language :: Python :: 3.8',
3638
'Programming Language :: Python :: 3.9',
3739
'Operating System :: OS Independent',
40+
'Topic :: Scientific/Engineering',
3841
'Topic :: Scientific/Engineering :: Artificial Intelligence',
42+
'Topic :: Software Development',
43+
'Topic :: Software Development :: Libraries',
44+
'Topic :: Software Development :: Libraries :: Python Modules',
3945
]
4046

4147
KEYWORDS: List[str] = sorted(

0 commit comments

Comments
 (0)