Minor refactoring for DelayDense/Sparse process (#693)

* Update process.py * Update process.py * Update process.py * Update process.py
lava-nc · May 16, 2023 · e39a4b6 · e39a4b6
1 parent 8576c19
commit e39a4b6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/lava/proc/dense/process.py b/src/lava/proc/dense/process.py
@@ -232,17 +232,18 @@ def __init__(self,
             spikes as binary spikes (num_message_bits = 0) or as graded
             spikes (num_message_bits > 0). Default is 0.
         """
+        if max_delay == 0:
+            max_delay = int(np.max(delays))
 
         super().__init__(weights=weights,
                          num_message_bits=num_message_bits,
                          name=name,
                          log_config=log_config,
+                         max_delay=max_delay,
                          **kwargs)
 
         self._validate_delays(weights, delays)
         shape = weights.shape
-        if max_delay == 0:
-            max_delay = int(np.max(delays))
 
         # Variables
         self.delays = Var(shape=shape, init=delays)

diff --git a/src/lava/proc/sparse/process.py b/src/lava/proc/sparse/process.py
@@ -245,17 +245,18 @@ def __init__(self,
             spikes as binary spikes (num_message_bits = 0) or as graded
             spikes (num_message_bits > 0). Default is 0.
         """
+        if max_delay == 0:
+            max_delay = int(np.max(delays))
 
         super().__init__(weights=weights,
                          num_message_bits=num_message_bits,
                          name=name,
                          log_config=log_config,
+                         max_delay=max_delay,
                          **kwargs)
 
         self._validate_delays(weights, delays)
         shape = weights.shape
-        if max_delay == 0:
-            max_delay = int(np.max(delays))
 
         # Variables
         self.delays = Var(shape=shape, init=delays)