Fix streaming py test for 1.0 APIs (#10520)

This commit is contained in:
Lixin Wei
2020-09-03 14:15:09 +08:00
committed by GitHub
parent ea95e6f7cc
commit 2f03bb5100
4 changed files with 7 additions and 7 deletions
+2 -2
View File
@@ -4,7 +4,7 @@ from ray.streaming import StreamingContext
def test_word_count():
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder() \
.build()
ctx.read_text_file(__file__) \
@@ -23,7 +23,7 @@ def test_word_count():
def test_simple_word_count():
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder() \
.build()
sink_file = "/tmp/ray_streaming_test_simple_word_count.txt"