Skip to content

Commit

Permalink
address flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
kengz committed Dec 30, 2020
1 parent e71ec95 commit 5170718
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torcharc/module_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def infer_in_shape(arc: dict, xs: Union[torch.Tensor, NamedTuple]) -> None:
arc.update(in_shape=in_shape)
elif nn_type == 'FiLMMerge':
assert ps.is_tuple(xs)
assert len(arc['in_names']) == 2, f'FiLMMerge in_names should only specify 2 keys for feature and conditioner'
assert len(arc['in_names']) == 2, 'FiLMMerge in_names should only specify 2 keys for feature and conditioner'
shapes = {name: list(x.shape)[1:] for name, x in xs._asdict().items() if name in arc['in_names']}
arc.update(shapes=shapes)
else:
Expand Down

0 comments on commit 5170718

Please sign in to comment.