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

fix typing
Some checks are pending
CI / lint_and_typecheck (push) Waiting to run
Details
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Details
Spell checker / Spell check (push) Waiting to run
Details

Browse source 
see https://github.com/microsoft/pyright/issues/9025
This commit is contained in:
Pierre Chapuis 2024-09-19 15:03:37 +02:00
parent cf270885a4
commit a542337a83
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -243,11 +243,15 @@ class Chain(ContextModule):
raise ChainError(message) from None raise ChainError(message) from None
def forward(self, *args: Any) -> Any: def forward(self, *args: Any) -> Any:
result: tuple[Any] | Any = None result: Any = None
intermediate_args: tuple[Any, ...] = args intermediate_args: tuple[Any, ...] = args
for name, layer in self._modules.items(): for name, layer in self._modules.items():
result = self._call_layer(layer, name, *intermediate_args) result = self._call_layer(layer, name, *intermediate_args)
intermediate_args = (result,) if not isinstance(result, tuple) else result if isinstance(result, tuple):
result = cast(tuple[Any], result)
intermediate_args = result
else:
intermediate_args = (result,)
self._reset_context() self._reset_context()
return result return result