Fix readme examples

janheinrichmerker · Jul 3, 2024 · 4e14766 · 4e14766
1 parent 6d93f75
commit 4e14766
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -38,7 +38,9 @@ from ray_elasticsearch import ElasticsearchDatasource
 
 init()
 source = ElasticsearchDslDatasource(index="test")
-res = read_datasource(source).sum("foo")
+res = read_datasource(source)\
+    .map(lambda x: x["_source"])\
+    .sum("id")
 print(f"Read complete. Sum: {res}")
 ```
 
@@ -72,7 +74,9 @@ from ray_elasticsearch import ElasticsearchDatasink
 
 init()
 sink = ElasticsearchDatasink(index="test")
-range(10_000).write_datasink(sink)
+range(10_000)\
+    .map(lambda x: {"_source": x})\
+    .write_datasink(sink)
 print("Write complete.")
 ```