UPDATE: Error Solved!
I found the cause of the error:
My fault, I put a decorator from dataclasses in front of the BaseModel, and this is what caused the issue.
@dataclass
class BaseRGCNHetero(nn.Module):
...

So for everybody else having this issue, just do not use the @dataclass decorator for more than one class.