Merge pull request #34 from autonomio/fix_array_shuffle

Fix array shuffle
autonomio · Apr 21, 2022 · bd87ebf · bd87ebf
2 parents 082b84d + 692eea8
commit bd87ebf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
 URL = 'http://autonom.io'
 LICENSE = 'MIT'
 DOWNLOAD_URL = 'https://github.com/autonomio/wrangle/'
-VERSION = '0.7.1'
+VERSION = '0.7.2'
 
 try:
     from setuptools import setup

diff --git a/wrangle/array/array_random_shuffle.py b/wrangle/array/array_random_shuffle.py
@@ -27,7 +27,7 @@ def array_random_shuffle(x, y=None, multi_input=False):
         for ar in x:
 
             rng.bit_generator.state = state
-            rng.shuffle(ar, axis=1)
+            rng.shuffle(ar, axis=0)
             x_out.append(ar)
 
         x = x_out
@@ -37,7 +37,7 @@ def array_random_shuffle(x, y=None, multi_input=False):
 
             try:
                 y.shape[1]
-                rng.shuffle(y, axis=1)
+                rng.shuffle(y, axis=0)
             except IndexError:
                 rng.shuffle(y)
 
@@ -50,14 +50,14 @@ def array_random_shuffle(x, y=None, multi_input=False):
     elif isinstance(x, list) == False:
 
         rng.bit_generator.state = state
-        rng.shuffle(x, axis=1)
+        rng.shuffle(x, axis=0)
 
         if y is not None:
             rng.bit_generator.state = state
 
             try:
                 y.shape[1]
-                rng.shuffle(y, axis=1)
+                rng.shuffle(y, axis=0)
             except IndexError:
                 rng.shuffle(y)