wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-03 22:42:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix streaming py test for 1.0 APIs (#10520)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
streaming/python/examples/wordcount.py
+1 -1
View File
@@ -65,7 +65,7 @@ if __name__ == "__main__":
args = parser.parse_args()
titles_file = str(args.titles_file)
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder() \
.option(Config.CHANNEL_TYPE, Config.NATIVE_CHANNEL) \
streaming/python/tests/test_stream.py
+3 -3
View File
@@ -3,7 +3,7 @@ from ray.streaming import StreamingContext
def test_data_stream():
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder().build()
stream = ctx.from_values(1, 2, 3)
java_stream = stream.as_java_stream()
@@ -17,7 +17,7 @@ def test_data_stream():
def test_key_data_stream():
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder().build()
key_stream = ctx.from_values(
"a", "b", "c").map(lambda x: (x, 1)).key_by(lambda x: x[0])
@@ -32,7 +32,7 @@ def test_key_data_stream():
def test_stream_config():
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder().build()
stream = ctx.from_values(1, 2, 3)
stream.with_config("k1", "v1")
streaming/python/tests/test_union_stream.py
+1 -1
View File
@@ -5,7 +5,7 @@ from ray.streaming import StreamingContext
def test_union_stream():
ray.init(load_code_from_local=True, include_java=True)
ray.init(_load_code_from_local=True, _include_java=True)
ctx = StreamingContext.Builder() \
.option("streaming.metrics.reporters", "") \
.build()
streaming/python/tests/test_word_count.py
+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"