Using sqlalchemy_utils ChoiceType not working

[im-ajaymeena]

Am I doing something wrong, I cant bring ChoiecType on admin page

model file

class COLLECTION_TYPES(enum.Enum):
    1 = '1'

class Collection(Base):
   id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
   name = Column(String, unique=True)
   user_id = Column("user", ForeignKey('user.id'))
   selected = Column(Boolean, default=False)

   created_at = Column(DateTime, default=datetime.datetime.now())
   updated_at = Column(DateTime, server_default=func.now(), onupdate=func.current_timestamp())
   user = relationship('User', backref='collection')
   num_entities = Column(Integer, default=0)


   collection_type = ChoiceType(COLLECTION_TYPES)
   encoded_meta_data = Column(String)

   @hybrid_property
   def milvus_collection_name(self) -> str:
       return f"_{self.id.replace('-', '_')}"

   __table_args__ = (UniqueConstraint(user_id, name),)



## alembic migration file (trimmed)

def upgrade() -> None:
    op.add_column('collection', sa.Column('collection_type', sqlalchemy_utils.types.choice.ChoiceType(COLLECTION_TYPES), nullable=True))

def downgrade() -> None:
    op.drop_column('collection', 'collection_type')

[aminalaee]

Hey @im-ajaymeena , what version of sqladmin are you using?

[im-ajaymeena]

@aminalaee
My version is the latest one for sqladmin

sqladmin 0.13.0
SQLAlchemy 2.0.19
SQLAlchemy-Utils 0.41.1

[aminalaee]

Yeah I think something wrong with your model definition: https://sqlalchemy-utils.readthedocs.io/en/latest/data_types.html#module-sqlalchemy_utils.types.choice
Why isn't the COLLECTION_TYPES used on the model?

[im-ajaymeena]

Actually its inside the model, sorry for incorrect indentation, I have updated the question
this is my actual code

[aminalaee]

I'm not much familiar with sqlalchemy_utils but why do you have this?

collection_type = ChoiceType(COLLECTION_TYPES)

But instead you should have:

collection_type = Column(ChoiceType(COLLECTION_TYPES))

[im-ajaymeena]

Oh My bad, Thanks a lot 🙏