From b5606b2999358d92e8c38cfbd40b621c64c04d42 Mon Sep 17 00:00:00 2001
From: nickfraser <icanlosh@gmail.com>
Date: Mon, 9 Sep 2024 16:28:33 +0100
Subject: [PATCH] Test (graph/quantize) Added extra prefix test to
 `layerwise_quantize` (#1022)

---
 tests/brevitas/graph/test_quantize.py | 5 +++++
 1 file changed, 5 insertions(+)

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': "<class 'torch.nn.modules.linear.Linear'>"},
+        {
+            'model': nn.Sequential(nn.Sequential(nn.Linear(2, 3))),
+            'name_blacklist': ['0'],
+            'key': '0.0',
+            'expected': "<class 'torch.nn.modules.linear.Linear'>"},
         {
             'model': nn.Sequential(nn.Sequential(nn.Linear(2, 3))),
             'name_blacklist': ['0.0'],