I described the issue in the PR that "fixes" it #8

There are problems after removing labels in running the code as is (to fix, labels needs to be remove from the input iterator; an empty labels should be sent to the model and then the OpenGNN library needs to deal with empty label dictionaries as opposed to None).