diff --git a/src/koza/io/writer/writer.py b/src/koza/io/writer/writer.py index c8b4f82..89ca6ff 100644 --- a/src/koza/io/writer/writer.py +++ b/src/koza/io/writer/writer.py @@ -20,6 +20,7 @@ def __init__( edge_properties: List[str] = None, sssom_config: SSSOMConfig = None, skip_checks: bool = False, + kwargs: dict = None, ): """Do not override this method; implement `init` instead.""" self.output_dir = output_dir @@ -30,7 +31,7 @@ def __init__( self.skip_checks = skip_checks self.converter = KGXConverter() - self.init() + self.init(**kwargs) def write(self, entities: Iterable): nodes, edges = self.converter.convert(entities) @@ -61,7 +62,7 @@ def check_extra_fields(row_keys: Tuple, columns: Tuple) -> None: raise ValueError(f"Extra fields found in row: {sorted(set(row_keys) - set(columns))}") @abstractmethod - def init(self): + def init(self, **kwargs): pass @abstractmethod