fix: add warning to non diff channel

jaxleyverse · Jan 10, 2025 · 9ff6309 · 9ff6309
1 parent 8b13502
commit 9ff6309
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jaxley/channels/lif.py b/jaxley/channels/lif.py
@@ -2,6 +2,7 @@
 # licensed under the Apache License Version 2.0, see <https://www.apache.org/licenses/>
 
 from typing import Dict, Optional
+from warnings import warn
 
 import jax.numpy as jnp
 
@@ -25,6 +26,8 @@ def __init__(self, name: Optional[str] = None):
         }
         self.current_name = f"{self.name}_i"
 
+        warn("This channel is not differentiable. Consider using SmoothLIF instead.")
+
     def update_states(self, states, dt, v, params):
         prefix = self._name
         # Decrease refractory time