update list of loaded layers after pruning

2 years ago · ddf1d54304
parent fd1de882ba
commit ddf1d54304
1 changed files with 1 additions and 0 deletions
--- 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