From e7627aa2b7b112cd19d885040839b1ffe1e12ecb Mon Sep 17 00:00:00 2001 From: TrisNol Date: Fri, 11 Aug 2023 16:31:58 +0200 Subject: [PATCH] refactor: Remove wrong typing from entities --- .../utils/postgres/entities.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/aki_prj23_transparenzregister/utils/postgres/entities.py b/src/aki_prj23_transparenzregister/utils/postgres/entities.py index 6d12edf..e249f77 100644 --- a/src/aki_prj23_transparenzregister/utils/postgres/entities.py +++ b/src/aki_prj23_transparenzregister/utils/postgres/entities.py @@ -101,7 +101,7 @@ class Sentiment(Base): # type: ignore company_hr = Column(Integer) company_court = Column(Integer) date = Column(DateTime(), default=datetime.now) - sentiment_type: SentimentTypeEnum = Column(Enum(SentimentTypeEnum), nullable=False) + sentiment_type = Column(Enum(SentimentTypeEnum), nullable=False) value = Column(Float(), nullable=False) source = Column(String(100)) @@ -150,7 +150,7 @@ class PersonRelation(Base): # type: ignore person_id = mapped_column(ForeignKey("person.id")) date_from = Column(DateTime(), default=datetime.now) date_to = Column(DateTime(), default=datetime.now) - relation: RelationTypeEnum = Column(Enum(RelationTypeEnum), nullable=False) + relation = Column(Enum(RelationTypeEnum), nullable=False) # company = relationship("Company") # person = relationship("Person", foreign_keys=[person_id]) @@ -182,9 +182,7 @@ class CompanyRelation(Base): # type: ignore company2_id = Column(Integer, nullable=False) date_from = Column(DateTime(), default=datetime.now) date_to = Column(DateTime(), default=datetime.now) - relation: RelationTypeCompanyEnum = Column( - Enum(RelationTypeCompanyEnum), nullable=False - ) + relation = Column(Enum(RelationTypeCompanyEnum), nullable=False) # company = relationship("Company")