0.1.4

kmheckel · Aug 21, 2023 · 2642cdf · 2642cdf
1 parent c904e9d
commit 2642cdf
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -23,7 +23,7 @@
 # This call to setup() does all the work
 setup(
     name="spyx",
-    version="0.1.3",
+    version="0.1.4",
     description="Spyx: SNNs in JAX",
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/spyx/_version.py b/spyx/_version.py
@@ -1 +1 @@
-__version__ = "0.1.3"
+__version__ = "0.1.4"
diff --git a/spyx/data.py b/spyx/data.py
@@ -59,7 +59,6 @@ class MNIST_loader(): # change this so that it just returns either rate or tempo
     # Change this to allow a config dictionary of 
     def __init__(self, time_steps=64, max_rate = 0.75, batch_size=32, val_size=0.3, subsample_data=1, key=jax.random.PRNGKey(0), download_dir='./MNIST'):
 
-        self.key = jax.random.PRNGKey(key)
         self.sample_T = time_steps
         self.max_rate = max_rate
         self.val_size = val_size