diff --git a/tests/brevitas/graph/test_quantize.py b/tests/brevitas/graph/test_quantize.py index 1d4ce0b67..45167b673 100644 --- a/tests/brevitas/graph/test_quantize.py +++ b/tests/brevitas/graph/test_quantize.py @@ -17,6 +17,11 @@ 'name_blacklist': ['0'], 'key': '0', 'expected': ""}, + { + 'model': nn.Sequential(nn.Sequential(nn.Linear(2, 3))), + 'name_blacklist': ['0'], + 'key': '0.0', + 'expected': ""}, { 'model': nn.Sequential(nn.Sequential(nn.Linear(2, 3))), 'name_blacklist': ['0.0'],