From 052d699ff31651f833a434a352f8247662cc4ac7 Mon Sep 17 00:00:00 2001 From: Michelle Ark Date: Thu, 21 Nov 2024 11:47:53 -0500 Subject: [PATCH] Implement MicrobatchModelRunner.__init__ --- core/dbt/task/run.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/core/dbt/task/run.py b/core/dbt/task/run.py index 0b991fc9f29..0c2888bb325 100644 --- 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