cleaned up lif, raf, wtas, fn, and quad-lif cells repr method

Author: Alexander Ororbia

ngclearn/components/neurons/spiking/LIFCell.py

@@ -1,5 +1,6 @@
 from ngclearn.components.jaxComponent import JaxComponent
 from jax import numpy as jnp, random, nn, Array
+from ngclearn.utils import tensorstats
 from ngclearn.utils.diffeq.ode_utils import get_integrator_code, \
                                             step_euler, step_rk2
 from ngclearn.utils.surrogate_fx import (secant_lif_estimator, arctan_estimator,
@@ -267,6 +268,19 @@ def help(cls): ## component help function
                 "hyperparameters": hyperparams}
         return info
 
+    def __repr__(self):
+        comps = [varname for varname in dir(self) if isinstance(getattr(self, varname), Compartment)]
+        maxlen = max(len(c) for c in comps) + 5
+        lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
+        for c in comps:
+            stats = tensorstats(getattr(self, c).value)
+            if stats is not None:
+                line = [f"{k}: {v}" for k, v in stats.items()]
+                line = ", ".join(line)
+            else:
+                line = "None"
+            lines += f"  {f'({c})'.ljust(maxlen)}{line}\n"
+        return lines
 
 if __name__ == '__main__':
     from ngcsimlib.context import Context

ngclearn/components/neurons/spiking/RAFCell.py

@@ -216,7 +216,7 @@ def help(cls): ## component help function
         return info
 
     def __repr__(self):
-        comps = [varname for varname in dir(self) if Compartment.is_compartment(getattr(self, varname))]
+        comps = [varname for varname in dir(self) if isinstance(getattr(self, varname), Compartment)]
         maxlen = max(len(c) for c in comps) + 5
         lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
         for c in comps:

ngclearn/components/neurons/spiking/WTASCell.py

@@ -160,11 +160,11 @@ def help(cls): ## component help function
         return info
 
     def __repr__(self):
-        comps = [varname for varname in dir(self) if Compartment.is_compartment(getattr(self, varname))]
+        comps = [varname for varname in dir(self) if isinstance(getattr(self, varname), Compartment)]
         maxlen = max(len(c) for c in comps) + 5
         lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
         for c in comps:
-            stats = tensorstats(getattr(self, c).get())
+            stats = tensorstats(getattr(self, c).value)
             if stats is not None:
                 line = [f"{k}: {v}" for k, v in stats.items()]
                 line = ", ".join(line)

ngclearn/components/neurons/spiking/adExCell.py

@@ -213,7 +213,7 @@ def help(cls): ## component help function
         return info
 
     def __repr__(self):
-        comps = [varname for varname in dir(self) if Compartment.is_compartment(getattr(self, varname))]
+        comps = [varname for varname in dir(self) if isinstance(getattr(self, varname), Compartment)]
         maxlen = max(len(c) for c in comps) + 5
         lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
         for c in comps:

ngclearn/components/neurons/spiking/fitzhughNagumoCell.py

@@ -215,7 +215,7 @@ def help(cls): ## component help function
         return info
 
     def __repr__(self):
-        comps = [varname for varname in dir(self) if Compartment.is_compartment(getattr(self, varname))]
+        comps = [varname for varname in dir(self) if isinstance(getattr(self, varname), Compartment)]
         maxlen = max(len(c) for c in comps) + 5
         lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
         for c in comps:

ngclearn/components/neurons/spiking/quadLIFCell.py

@@ -242,7 +242,7 @@ def help(cls): ## component help function
         return info
 
     def __repr__(self):
-        comps = [varname for varname in dir(self) if Compartment.is_compartment(getattr(self, varname))]
+        comps = [varname for varname in dir(self) if isinstance(getattr(self, varname), Compartment)]
         maxlen = max(len(c) for c in comps) + 5
         lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
         for c in comps: