mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

use find_packages() in setup.py

Browse Source
This commit is contained in:
Matthew Johnson 2018-12-11 15:45:56 -08:00
parent 8fac32e505
commit 3ac1001c49
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
setup.py
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from setuptools import setup
from setuptools import setup, find_packages
setup(
name='jax',
@ -20,8 +20,7 @@ setup(
description='Differentiate, compile, and transform Numpy code.',
author='JAX team',
author_email='jax-dev@google.com',
packages=['jax', 'jax.lib', 'jax.interpreters', 'jax.numpy', 'jax.scipy',
'jax.scipy.stats', 'jax.experimental'],
packages=find_packages(),
install_requires=['numpy>=1.12', 'six', 'protobuf>=3.6.0', 'absl-py',
'opt_einsum'],
url='https://github.com/google/jax',