Skip to content

Commit 82fbe57

Browse files
Include missing dependencies in setup.py (#3611) (#3650)
Co-authored-by: Will Cromar <[email protected]>
1 parent 60d39ee commit 82fbe57

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

setup.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
third_party_path = os.path.join(base_dir, 'third_party')
5555

5656
_libtpu_version = '0.1.dev20220518'
57-
_litbpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
57+
_libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
5858

5959

6060
def _get_build_mode():
@@ -354,10 +354,14 @@ def make_relative_rpath(path):
354354
[make_relative_rpath('torch_xla/lib')],
355355
),
356356
],
357+
install_requires=[
358+
'absl-py>=1.0.0',
359+
'cloud-tpu-client>=0.10.0',
360+
],
357361
extras_require={
358362
# On Cloud TPU VM install with:
359363
# $ sudo pip3 install torch_xla[tpuvm] -f https://storage.googleapis.com/tpu-pytorch/wheels/tpuvm/torch_xla-1.11-cp38-cp38-linux_x86_64.whl
360-
'tpuvm': [f'libtpu-nightly @ {_litbpu_storage_path}'],
364+
'tpuvm': [f'libtpu-nightly @ {_libtpu_storage_path}'],
361365
},
362366
package_data={
363367
'torch_xla': [

0 commit comments

Comments
 (0)