Update models.py (#696)

lava-nc · May 17, 2023 · 4beccfa · 4beccfa
1 parent e39a4b6
commit 4beccfa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/lava/proc/sparse/models.py b/src/lava/proc/sparse/models.py
@@ -245,6 +245,9 @@ def get_delay_wgts_mat(weights, delays) -> spmatrix:
         weights.
         """
         # Can only start at 1, as delays==0 raises inefficiency warning
+        if np.max(delays) == 0:
+            return weights
+
         weight_delay_from_1 = vstack([weights.multiply(delays == k)
                                       for k in range(1, np.max(delays) + 1)])
         # Create weight matrix at delays == 0