Unit tests are broken
It seems like the unit tests are out of date.
-
test_gnn_model.py
seems to be looking for a modulegnn_stage
that does not exist instages.edge_classifier
-
test_gnn_stage.py
fails because thetest_gnn_config.yaml
does not have the keymodel