From ddf1d543047da866e03d093d17bebb97a35615fd Mon Sep 17 00:00:00 2001
From: Jan Petykiewicz <anewusername@gmail.com>
Date: Tue, 5 Apr 2022 18:33:22 -0700
Subject: [PATCH] update list of loaded layers after pruning

---
 snarled/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/snarled/main.py b/snarled/main.py
index 047ac86..7c04a75 100644
--- a/snarled/main.py
+++ b/snarled/main.py
@@ -155,6 +155,7 @@ def trace_connectivity(
         finished_layers = loaded_layers - remaining_layers
         for layer in finished_layers:
             nets_info.prune(layer)
+            loaded_layers.remove(layer)
 
     return nets_info