diff --git a/deeplabcut/generate_training_dataset/trainingsetmanipulation.py b/deeplabcut/generate_training_dataset/trainingsetmanipulation.py index a51826931e..623b336a80 100755 --- a/deeplabcut/generate_training_dataset/trainingsetmanipulation.py +++ b/deeplabcut/generate_training_dataset/trainingsetmanipulation.py @@ -845,7 +845,10 @@ def create_training_dataset( ) create_multianimaltraining_dataset( - config, num_shuffles, Shuffles, net_type=net_type + config, num_shuffles, Shuffles, + net_type=net_type, + trainIndices=trainIndices, + testIndices=testIndices, ) else: scorer = cfg["scorer"]