Implement MicrobatchModelRunner.__init__

dbt-labs · Nov 21, 2024 · 052d699 · 052d699
1 parent 255eaf2
commit 052d699
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/core/dbt/task/run.py b/core/dbt/task/run.py
@@ -334,9 +334,12 @@ def execute(self, model, manifest):
 
 
 class MicrobatchModelRunner(ModelRunner):
-    batch_idx: Optional[int] = None
-    batches: Dict[int, BatchType] = {}
-    relation_exists: bool = False
+    def __init__(self, config, adapter, node, node_index: int, num_nodes: int):
+        super().__init__(config, adapter, node, node_index, num_nodes)
+
+        self.batch_idx: Optional[int] = None
+        self.batches: Dict[int, BatchType] = {}
+        self.relation_exists: bool = False
 
     def set_batch_idx(self, batch_idx: int) -> None:
         self.batch_idx = batch_idx