Fix the issue of the filter in select not working in the listener.

superduper-io · Jul 5, 2024 · 064ed9b · 064ed9b
1 parent c5afaca
commit 064ed9b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/superduperdb/base/datalayer.py b/superduperdb/base/datalayer.py
@@ -741,16 +741,18 @@ def _build_task_workflow(
             return G
 
         for listener in listeners:
+            from superduperdb import Listener
+            listener = t.cast(Listener, listener)
             G.add_node(
                 f'{listener.model.identifier}.predict_in_db({listener.uuid})',
                 job=ComponentJob(
                     component_identifier=listener.model.identifier,
                     kwargs={
                         'X': listener.key,
                         'ids': ids,
-                        'select': query.encode(),
+                        'select': listener.select.encode(),
                         'overwrite': overwrite,
-                        'predict_id': listener.uuid,
+                        'predict_id': listener.predict_id,
                         **listener.predict_kwargs,
                     },
                     method_name='predict_in_db',