Correct signature for abstractish base methods.

equinor · Dec 4, 2023 · dc1ac7f · dc1ac7f
1 parent 0381cc1
commit dc1ac7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fmu/sumo/explorer/objects/_document_collection.py b/src/fmu/sumo/explorer/objects/_document_collection.py
@@ -245,10 +245,10 @@ async def _next_batch_async(self) -> List[Dict]:
 
         return len(hits["hits"])
 
-    def _postprocess_batch(self, hits):
+    def _postprocess_batch(self, hits, pit):
         return
 
-    async def _postprocess_batch_async(self, hits):
+    async def _postprocess_batch_async(self, hits, pit):
         return
 
     def _init_query(self, doc_type: str, query: Dict = None) -> Dict: