# RockMapper if seg_model in ['RockMapper_20250628_v1', 'RockMapper_20251117_v2']: from rockmapper.rock_mapper import do_work