-
Notifications
You must be signed in to change notification settings - Fork 16
/
setup.py
49 lines (41 loc) · 1.18 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
"""Setup.py file for array_record."""
from setuptools import find_packages
from setuptools import setup
from setuptools.dist import Distribution
REQUIRED_PACKAGES = [
'absl-py',
'etils[epath]',
]
BEAM_EXTRAS = [
'apache-beam[gcp]==2.53.0',
'google-cloud-storage>=2.11.0',
'tensorflow>=2.14.0'
]
class BinaryDistribution(Distribution):
"""This class makes 'bdist_wheel' include an ABI tag on the wheel."""
def has_ext_modules(self):
return True
setup(
name='array_record',
version='0.5.1',
description='A file format that achieves a new frontier of IO efficiency',
author='ArrayRecord team',
author_email='no-reply@google.com',
packages=find_packages(),
include_package_data=True,
package_data={'': ['*.so']},
python_requires='>=3.9',
install_requires=REQUIRED_PACKAGES,
extras_require={
'beam': BEAM_EXTRAS
},
url='https://github.com/google/array_record',
license='Apache-2.0',
classifiers=[
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
],
zip_safe=False,
distclass=BinaryDistribution,
)