Explicit axis keyword

tbepler · Nov 20, 2023 · 25cb2cb · 25cb2cb
1 parent dadfe72
commit 25cb2cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/topaz/commands/train_test_split.py b/topaz/commands/train_test_split.py
@@ -109,8 +109,8 @@ def main(args):
         image_names_train.append(image_names[j])
         groups_train.append(groups[j])
 
-    targets_train = pd.concat(groups_train, 0)
-    targets_test = pd.concat(groups_test, 0)
+    targets_train = pd.concat(groups_train, axis=0)
+    targets_test = pd.concat(groups_test, axis=0)
 
 
     ## if the image-dir is specified, make the image list files