Skip to content

Commit fe87aad

Browse files
committed
Update setup.py to now support multiple modules
1 parent 7ff5b7a commit fe87aad

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

setup.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,11 @@
99
name="gradient-centralization-tf",
1010
version="0.0.1",
1111
description="Implement Gradient Centralization in TensorFlow",
12-
py_modules=["gctf"],
12+
packages=["gctf"],
13+
1314
long_description=long_description,
1415
long_description_content_type="text/markdown",
16+
1517
classifiers=[
1618
"Programming Language :: Python :: 3",
1719
"Programming Language :: Python :: 3.7",
@@ -20,13 +22,16 @@
2022
"Topic :: Scientific/Engineering :: Artificial Intelligence",
2123
"License :: OSI Approved :: Apache Software License",
2224
],
25+
2326
url="https://github.com/Rishit-dagli/Gradient-Centralization-TensorFlow",
2427
author="Rishit Dagli",
2528
author_email="[email protected]",
29+
2630
install_requires=[
2731
"tensorflow ~= 2.4.0",
2832
"keras ~= 2.4.0",
2933
],
34+
3035
extras_require={
3136
"dev": [
3237
"check-manifest",

0 commit comments

Comments
 (0)