(minor) Fix Flake8.

This commit is contained in:
Ryan Dick 2023-10-10 10:06:53 -04:00
parent 61242bf86a
commit f3c138a208

View File

@ -67,4 +67,4 @@ def test_skip_torch_weight_init_restores_base_class_behavior():
_ = torch.nn.Conv1d(10, 20, 3)
torch.nn.modules.conv._ConvNd.reset_parameters = saved_fn
assert called_monkey_patched_fn == True
assert called_monkey_patched_fn