github.com/apache/beam/sdks/v2@v2.48.2/python/apache_beam/testing/benchmarks/chicago_taxi/setup.py (about) 1 # Copyright 2019 Google LLC. All Rights Reserved. 2 # 3 # Licensed under the Apache License, Version 2.0 (the "License"); 4 # you may not use this file except in compliance with the License. 5 # You may obtain a copy of the License at 6 # 7 # https://www.apache.org/licenses/LICENSE-2.0 8 # 9 # Unless required by applicable law or agreed to in writing, software 10 # distributed under the License is distributed on an "AS IS" BASIS, 11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 # See the License for the specific language governing permissions and 13 # limitations under the License. 14 15 """Setup dependencies for local and cloud deployment.""" 16 # pytype: skip-file 17 18 import setuptools 19 20 TF_VERSION = '1.14.0' 21 22 if __name__ == '__main__': 23 setuptools.setup( 24 name='beam_chicago_taxi', 25 version='0.13.0', 26 packages=setuptools.find_packages(), 27 install_requires=[ 28 'jupyter>=1.0,<2', 29 'notebook>=5.7.8,<5.8', 30 'tensorflow==' + TF_VERSION, 31 'tensorflow-data-validation>=0.13.1,<0.14', 32 'tensorflow-metadata>=0.13.0,<0.14', 33 'tensorflow-model-analysis>=0.13.2,<0.14', 34 'tensorflow-serving-api>=1.13.0,<1.14', 35 'tensorflow-transform>=0.13.0,<0.14', 36 ], 37 python_requires='>=2.7,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,<4', 38 )