Update base_gefx_generator.py to new pipeline