From 373c1183a3e65a98950efe84d5537f3c52e6762a Mon Sep 17 00:00:00 2001
From: amitjans <amitjans@ethz.ch>
Date: Tue, 29 Jun 2021 18:37:51 +0200
Subject: [PATCH] small fix
---
tools.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/tools.py b/tools.py
index 73ba1c0..95d0513 100644
--- a/tools.py
+++ b/tools.py
@@ -246,17 +246,16 @@ class Memory:
if self.i == settings.NUM_ITERATIONS:
list_len = []
- count_copy = self.count.copy()
- for i, val in enumerate(count_copy):
+ for i, val in enumerate(self.count):
if 0 in val:
val.remove(0)
if len(val) == 0:
del self.memory[i]
- del self.count[i]
else:
val = [val[i] - 1 for i in range(len(val))]
self.count[i] = val
list_len.append(len(val))
+ self.count = [i for i in self.count if len(i) > 0]
self.i -= 1
self.max_idx = np.argmax(list_len)
--
GitLab