Bug fix

usedToBeTomas · Jul 14, 2023 · 939aed6 · 939aed6
1 parent 584af92
commit 939aed6
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/examples/1. simple_image_classifier/test1.npz b/examples/1. simple_image_classifier/test1.npz
diff --git a/examples/2. mnist/mnist.npz b/examples/2. mnist/mnist.npz
diff --git a/pod.py b/pod.py
@@ -51,6 +51,6 @@ def load(**options):
             if options.get("one_hot") != None:
                 Q =  []
                 for i in range(len(data)):
-                    Q.append(decimal_to_array(data[i], int(options.get("one-hot"))))
+                    Q.append(decimal_to_array(data[i], int(options.get("one_hot"))))
                 data = np.vstack(Q)
             return data
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name='pileoffeather',
-    version='0.3.0',
+    version='0.3.1',
     license='MIT',
     url = 'https://github.com/usedToBeTomas/pile-of-feather',
     author='Daniele Tomaselli',