Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Distribute: improve sanity check error message

Browse source 
E.g.:

    AssertionError: Number of positional arguments (1) must match number of sub-modules (2).
This commit is contained in:
Cédric Deltheil 2023-09-28 11:40:56 +02:00 committed by Cédric Deltheil
parent 620e58a593
commit 88e454f1cb
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -411,7 +411,8 @@ class Distribute(Chain):
_tag = "DISTR"
def forward(self, *args: Any) -> tuple[Tensor, ...]:
assert len(args) == len(self._modules), "Number of positional arguments must match number of sub-modules."
n, m = len(args), len(self._modules)
assert n == m, f"Number of positional arguments ({n}) must match number of sub-modules ({m})."
return tuple([self.call_layer(module, name, arg) for arg, (name, module) in zip(args, self._modules.items())])
def _show_only_tag(self) -> bool: