get_training_variants.py: minor changes