refactor: optimized graph loader (#290)

Co-authored-by: anna-grim <anna.grim@alleninstitute.org>

Author: anna-grim