from dataset_generation.gridnet_hd_batch_generation import batch_generation_main # Example: # python scripts/run_batch_generation.py --config configs/config_gridnet_hd_dataset_generation.yaml batch_generation_main()