Merge pull request #55 from spectriclabs/dev

Dev -> master
spectriclabs · May 31, 2024 · 2dd20f8 · 2dd20f8
2 parents 9f0fa9e + 52b11a5
commit 2dd20f8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/elastic_datashader/elastic.py b/elastic_datashader/elastic.py
@@ -35,13 +35,13 @@ def scan(search, use_scroll=False, size=10000):
     search = search.sort("_doc")
     if use_scroll:
         for hit in search.scan():
-            yield hit
+            yield from hit
     else:
         _search = search.params(size=size).extra(track_total_hits=False)
         while _search is not None:
             hit = None
             for hit in _search:
-                yield hit
+                yield from hit
             if hit is not None:
                 _search = search.extra(search_after=list(hit.meta.sort))
             else:
@@ -597,7 +597,7 @@ def run_search(**kwargs):
 
         while response.aggregations.comp.buckets:
             for b in response.aggregations.comp.buckets:
-                yield b
+                yield from b
             if "after_key" in response.aggregations.comp:
                 after = response.aggregations.comp.after_key
             else:

diff --git a/elastic_datashader/tilegen.py b/elastic_datashader/tilegen.py
@@ -778,6 +778,8 @@ def generate_nonaggregated_tile(
             # now for the points as well
             points_agg = None
 
+            points_color_key = None
+
             if df_points is not None:
                 df_points["c"] = df_points["c"].astype("category")
                 # prevent memory explosion in datashader _colorize