graphene-sqlalchemy VS wagtail-grapple

I have a project in python 2.7 and started migrating it to 3.9, after some upgrades I started getting this error "AttributeError: entity" similar to this issue. I already googled similar issues and I think it has to do with a relationship or table class model. Is there a way I can check if my table class models and relationships created properly.

Before we add the other objects to this file, let’s discuss what we have so far. We imported our models from our models.py file. Then we are creating our class UserObject, which we can see extends the SQLAlchemyObjectType – a class provided by graphene-sqlalchemy to integrate SQLAlchemy and GraphQL. The user_id attribute tells graphene to use the id associated with the model in the database, rather than the unique id graphql provides. You don’t need to use this attribute if you don’t want to, I find it useful for my applications. Finally, we are classing the inner class Meta that allows us to set different options. We are telling graphene_sqlalchemy to use the UserModel and to use the Relay Node.