wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-10 12:38:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Type check ObjectRef (#9856)

Browse Source 
* Type check ObjectRef

* Bug fix

* Port typing tests to bazel test
This commit is contained in:
Simon Mo
2020-08-11 10:38:29 -07:00
committed by GitHub
parent 98df612010
commit 4c52adddfa
8 changed files with 119 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ci/travis/check_typing_bad.py
-25
View File
@@ -1,25 +0,0 @@
import ray
ray.init()
@ray.remote
def f(a: int) -> str:
return "a = {}".format(a + 1)
@ray.remote
def g(s: str) -> str:
return s + " world"
@ray.remote
def h(a: str, b: int) -> str:
return a
# Does not typecheck:
a = h.remote(1, 1)
b = f.remote("hello")
c = f.remote(1, 1)
d = f.remote(1) + 1
ci/travis/check_typing_good.py
-26
View File
@@ -1,26 +0,0 @@
import ray
ray.init()
@ray.remote
def f(a: int) -> str:
return "a = {}".format(a + 1)
@ray.remote
def g(s: str) -> str:
return s + " world"
@ray.remote
def h(a: str, b: int) -> str:
return a
print(f.remote(1))
x = f.remote(1)
print(g.remote(x))
# typechecks but doesn't run
print(ray.get(f.remote(x)))