Refactor Module Map to have a single build_graph function inside for-loop

This is partly for readability, but partly for the inference script: Now we can call module_map_model.build_graph(...) on a single event and it will perform inference.