monkey patch scatter

xpdAcq · Sep 19, 2018 · 2532ed8 · 2532ed8
1 parent b14a898
commit 2532ed8
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/news/scatter_mp b/news/scatter_mp
@@ -0,0 +1,13 @@
+**Added:** None
+
+**Changed:** None
+
+**Deprecated:** None
+
+**Removed:** None
+
+**Fixed:**
+
+* Monkey patch ``Stream.scatter`` to use the parallel backend
+
+**Security:** None
diff --git a/streamz_ext/core.py b/streamz_ext/core.py
@@ -11,6 +11,12 @@
 from streamz.core import _global_sinks, _truthy
 
 
+def scatter(self, **kwargs):
+    from .parallel import scatter
+    return scatter(self, **kwargs)
+
+Stream.scatter = scatter
+
 @Stream.register_api()
 class starsink(Stream):
     """ Apply a function on every element