Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 23:12:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

simplify Chain#append

Browse source
This commit is contained in:
Pierre Chapuis 2023-08-23 17:12:54 +02:00
parent beacfe816b
commit 802970e79a
1 changed files with 1 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/refiners/fluxion/layers/chain.py
View file

@ -337,12 +337,7 @@ class Chain(ContextModule):
raise ValueError(f"No module of type {module_type.__name__} found in the chain.") raise ValueError(f"No module of type {module_type.__name__} found in the chain.")
def append(self, module: Module) -> None: def append(self, module: Module) -> None:
modules = list(self) self.insert(-1, module)
modules.append(module)
self._regenerate_keys(modules)
if isinstance(module, ContextModule):
module._set_parent(self)
self._register_provider()
def pop(self, index: int = -1) -> Module | tuple[Module]: def pop(self, index: int = -1) -> Module | tuple[Module]:
modules = list(self) modules = list(self)