From 0d84e6a7e666717cabeca7d205b021e541e268aa Mon Sep 17 00:00:00 2001
From: Darnell Granberry <40174000+DarnellGranberry@users.noreply.github.com>
Date: Fri, 13 Dec 2024 17:24:53 -0500
Subject: [PATCH] fix pd concat axis=0 in 1 more place

---
 topaz/utils/data/train_test_split_micrographs.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/topaz/utils/data/train_test_split_micrographs.py b/topaz/utils/data/train_test_split_micrographs.py
index fa9f7de..5a5a1f4 100644
--- a/topaz/utils/data/train_test_split_micrographs.py
+++ b/topaz/utils/data/train_test_split_micrographs.py
@@ -40,8 +40,8 @@ def train_test_split_micrographs(seed, n, path, format, image_dir, file_ext):
         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