Include missing dependencies in setup.py (#3611) (#3650)

Co-authored-by: Will Cromar <[email protected]>
pytorch · Jun 14, 2022 · 82fbe57 · 82fbe57
1 parent 60d39ee
commit 82fbe57
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -54,7 +54,7 @@
 third_party_path = os.path.join(base_dir, 'third_party')
 
 _libtpu_version = '0.1.dev20220518'
-_litbpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
+_libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
 
 
 def _get_build_mode():
@@ -354,10 +354,14 @@ def make_relative_rpath(path):
                 [make_relative_rpath('torch_xla/lib')],
         ),
     ],
+    install_requires=[
+      'absl-py>=1.0.0',
+      'cloud-tpu-client>=0.10.0',
+    ],
     extras_require={
         # On Cloud TPU VM install with:
         # $ 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
-        'tpuvm': [f'libtpu-nightly @ {_litbpu_storage_path}'],
+        'tpuvm': [f'libtpu-nightly @ {_libtpu_storage_path}'],
     },
     package_data={
         'torch_xla': [