diff --git a/tests/test_flop_count.py b/tests/test_flop_count.py index ad593b9..8ab7eeb 100644 --- a/tests/test_flop_count.py +++ b/tests/test_flop_count.py @@ -189,6 +189,7 @@ def test_customized_ops(self) -> None: The second case checks when a new handle for a default operation is passed. The new handle should overwrite the default handle. """ + # New handle for a new operation. def dummy_sigmoid_flop_jit( inputs: typing.List[Any], outputs: typing.List[Any]