Upload 2 files

snntorch/__init__.py

@@ -0,0 +1,27 @@
+"""
+Stub implementation for snntorch package.
+"""
+import logging
+import torch
+
+from ._neurons import LIF, Synaptic
+
+logger = logging.getLogger(__name__)
+logger.info("Using minimal stub implementation of snntorch")
+
+# Spike generation functions used in communicator.py
+class spikegen:
+    @staticmethod
+    def rate(tensor, num_steps=1, threshold=None):
+        """Generate spikes from tensor"""
+        if threshold is None:
+            threshold = 0.5
+            
+        # Simple thresholding to generate spikes
+        spikes = (tensor > threshold).float()
+        return spikes
+        
+    @staticmethod
+    def latency(tensor, num_steps=None, threshold=None, tau=1.0):
+        """Generate spikes with latency coding"""
+        return spikegen.rate(tensor, num_steps, threshold)

snntorch/_neurons.py

@@ -0,0 +1,14 @@
+"""
+Stub implementation of the snntorch._neurons module
+"""
+import torch
+
+class Synaptic(torch.nn.Module):
+    """Stub Synaptic neuron class"""
+    def __init__(self, alpha=0.1, beta=0.9):
+        super().__init__()
+        self.alpha = alpha
+        self.beta = beta
+        
+    def forward(self, x):
+        return x