kwargs for initializer

edikedik · Mar 13, 2024 · e322291 · e322291
1 parent 9f9d17a
commit e322291
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/lXtractor/collection/constructor.py b/lXtractor/collection/constructor.py
@@ -517,7 +517,9 @@ def fetch_missing(self, items: abc.Iterable[SeqItem]) -> t.Any:
 
     def init_inputs(self, items: SeqItemList) -> lxc.ChainList[lxc.ChainSequence]:
         staged = items.prep_for_init(self.paths)
-        chains = self.interfaces.Initializer.from_iterable(staged)
+        chains = self.interfaces.Initializer.from_iterable(
+            staged, **self.config["Initializer_init_kw"]
+        )
         return lxc.ChainList(filter(lambda x: x is not None, chains))
 
 
@@ -560,7 +562,9 @@ def fetch_missing(self, items: abc.Iterable[StrItem]) -> t.Any:
 
     def init_inputs(self, items: StrItemList) -> lxc.ChainList[lxc.ChainStructure]:
         staged = items.prep_for_init(self.paths)
-        chains = self.interfaces.Initializer.from_iterable(staged)
+        chains = self.interfaces.Initializer.from_iterable(
+            staged, **self.config["Initializer_init_kw"]
+        )
         chains = filter(lambda x: x is not None, chains)
         return lxc.ChainList(chain.from_iterable(chains))
 
@@ -595,7 +599,8 @@ def fetch_missing(self, items: abc.Iterable[MapItem]) -> tuple[t.Any, t.Any]:
     def init_inputs(self, items: MapItemList) -> lxc.ChainList[lxc.Chain]:
         return lxc.ChainList(
             self.interfaces.Initializer.from_mapping(
-                dict(items.prep_for_init(self.paths))
+                dict(items.prep_for_init(self.paths)),
+                **self.config["Initializer_init_kw"],
             )
         )
 

diff --git a/lXtractor/resources/collection_config.json b/lXtractor/resources/collection_config.json
@@ -25,6 +25,7 @@
   "SIFTS_kwargs": {},
   "UniProt_kwargs": {},
   "Initializer_kwargs": {},
+  "Initializer_init_kw": {},
   "write_batches": false,
   "write_batches_overwrite": true,
   "keep_chains": false,