From 9ff63091d810afcbf172e67836c6193121d05b34 Mon Sep 17 00:00:00 2001 From: jnsbck-uni Date: Fri, 10 Jan 2025 23:13:41 +0100 Subject: [PATCH] fix: add warning to non diff channel --- jaxley/channels/lif.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/jaxley/channels/lif.py b/jaxley/channels/lif.py index 1b491a3a..b99f2f4c 100644 --- a/jaxley/channels/lif.py +++ b/jaxley/channels/lif.py @@ -2,6 +2,7 @@ # licensed under the Apache License Version 2.0, see 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