Update action_dag_diagram script to work with new config.