Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 14:18:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

always respect _can_refresh_parent

Browse source
This commit is contained in:
Pierre Chapuis 2023-09-01 15:10:07 +02:00
parent d389d11a06
commit bd59790e08
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -142,7 +142,7 @@ class Chain(ContextModule):
self._reset_context() self._reset_context()
for module in self: for module in self:
if isinstance(module, ContextModule) and module._can_refresh_parent and module.parent != self: if isinstance(module, ContextModule) and module.parent != self:
module._set_parent(self) module._set_parent(self)
@property @property

2
src/refiners/fluxion/layers/module.py
View file

@ -62,6 +62,8 @@ class ContextModule(Module):
return self._parent[0] return self._parent[0]
def _set_parent(self, parent: "Chain | None") -> None: def _set_parent(self, parent: "Chain | None") -> None:
if not self._can_refresh_parent:
return
if parent is None: if parent is None:
self._parent = [] self._parent = []
return return