From 5170718f717d485305da7ed0438132130240887a Mon Sep 17 00:00:00 2001 From: kengz Date: Tue, 29 Dec 2020 22:07:43 -0500 Subject: [PATCH] address flake8 --- torcharc/module_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torcharc/module_builder.py b/torcharc/module_builder.py index 3f4f494..dee25c6 100644 --- a/torcharc/module_builder.py +++ b/torcharc/module_builder.py @@ -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: