Skip to content

Commit 0d84e6a

Browse files
fix pd concat axis=0 in 1 more place
1 parent 267ab51 commit 0d84e6a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

topaz/utils/data/train_test_split_micrographs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ def train_test_split_micrographs(seed, n, path, format, image_dir, file_ext):
4040
image_names_train.append(image_names[j])
4141
groups_train.append(groups[j])
4242

43-
targets_train = pd.concat(groups_train, 0)
44-
targets_test = pd.concat(groups_test, 0)
43+
targets_train = pd.concat(groups_train, axis=0)
44+
targets_test = pd.concat(groups_test, axis=0)
4545

4646

4747
## if the image-dir is specified, make the image list files

0 commit comments

Comments
 (0)