Update tpu-test.py

Author: richardsliu

applications/ray/kuberay-tpu-webhook/tests/tpu-test.py

@@ -1,6 +1,6 @@
 import ray
 
-@ray.remote(resources={"google.com/tpu": 4})
+@ray.remote(resources={"TPU": 4})
 def tpu_cores():
     import jax
 
@@ -23,4 +23,4 @@ def tpu_cores():
 
 num_workers = 4
 result = [tpu_cores.remote() for _ in range(num_workers)]
-print(ray.get(result))
+print(ray.get(result))