Skip to content

Commit

Permalink
Add torch to reqs that are not autoinstalled and format setup.py. (#116)
Browse files Browse the repository at this point in the history
  • Loading branch information
kboyd committed Apr 14, 2022
1 parent ca729e9 commit 9eacab5
Showing 1 changed file with 16 additions and 20 deletions.
36 changes: 16 additions & 20 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

this_dir = path.abspath(path.dirname(__file__))

with open(path.join(this_dir, 'README.md'), encoding='utf-8') as f:
with open(path.join(this_dir, "README.md"), encoding="utf-8") as f:
long_description = f.read()


Expand All @@ -28,39 +28,35 @@ def reqs(file, without=None):
]


compat_reqs = [
'dataclasses==0.7;python_version<"3.7"'
]
compat_reqs = ['dataclasses==0.7;python_version<"3.7"']

utils_reqs = reqs("utils-requirements.txt")
test_reqs = reqs("test-requirements.txt")

setup(
name='gretel-synthetics',
author='Gretel Labs, Inc.',
author_email='open-source@gretel.ai',
name="gretel-synthetics",
author="Gretel Labs, Inc.",
author_email="open-source@gretel.ai",
use_scm_version=True,
setup_requires=["setuptools_scm"],
description='Synthetic Data Generation with optional Differential Privacy',
url='https://github.com/gretelai/gretel-synthetics',
license='http://www.apache.org/licenses/LICENSE-2.0',
description="Synthetic Data Generation with optional Differential Privacy",
url="https://github.com/gretelai/gretel-synthetics",
license="http://www.apache.org/licenses/LICENSE-2.0",
long_description=long_description,
long_description_content_type='text/markdown',
package_dir={'': 'src'},
packages=find_packages('src'),
long_description_content_type="text/markdown",
package_dir={"": "src"},
packages=find_packages("src"),
python_requires=">=3.6",
install_requires=reqs("requirements.txt", without=["tensorflow=="]) + compat_reqs,
extras_require={
'all': utils_reqs,
'utils': utils_reqs,
'test': test_reqs
},
install_requires=(
reqs("requirements.txt", without=["tensorflow==", "torch=="]) + compat_reqs
),
extras_require={"all": utils_reqs, "utils": utils_reqs, "test": test_reqs},
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Operating System :: POSIX :: Linux",
"Operating System :: MacOS",
"Operating System :: Microsoft :: Windows",
"Topic :: Scientific/Engineering :: Artificial Intelligence"
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
)

0 comments on commit 9eacab5

Please sign in to comment.