Error loading the ogb-molpcba dataset

[AbdoSharaf98]

In ogbmolpcba_dataset.py line 96 (and similarly 98), a PyGCollater object is initialized without passing a required positional argument (dataset) which raises an error when calling the get_dataset function for the molecule dataset. I think this can be fixed by replacing with self._collate = PyGCollater(self.ogb_dataset, follow_batch=[], exclude_keys=[])? Or is this how it is supposed to be and I am missing something?