Skip to content

Commit 0484cf5

Browse files
authored
Minor type hint fixes (#1488)
* Minor type hint fixes * Fix
1 parent c030c9c commit 0484cf5

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tortoise/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def __init__(self, meta: "Model.Meta") -> None:
244244
self.basequery_all_fields: Query = Query()
245245
self.basetable: Table = Table("")
246246
self.pk_attr: str = getattr(meta, "pk_attr", "")
247-
self.generated_db_fields: Tuple[str] = None # type: ignore
247+
self.generated_db_fields: Tuple[str, ...] = None # type: ignore
248248
self._model: Type["Model"] = None # type: ignore
249249
self.table_description: str = getattr(meta, "table_description", "")
250250
self.pk: Field = None # type: ignore
@@ -329,7 +329,7 @@ def finalise_fields(self) -> None:
329329
if not field.generated:
330330
continue
331331
generated_fields.append(field.source_field or field.model_field_name)
332-
self.generated_db_fields = tuple(generated_fields) # type: ignore
332+
self.generated_db_fields = tuple(generated_fields)
333333

334334
self._ordering_validated = True
335335
for field_name, _ in self._default_ordering:

tortoise/queryset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def sql(self, **kwargs) -> str:
253253
"""Return the actual SQL."""
254254
return self.as_query().get_sql(**kwargs)
255255

256-
def as_query(self) -> QUERY:
256+
def as_query(self) -> QueryBuilder:
257257
"""Return the actual query."""
258258
if self._db is None:
259259
self._db = self._choose_db() # type: ignore
@@ -1700,7 +1700,7 @@ def __init__(
17001700
self.objects = objects
17011701
self.fields = fields
17021702
self.batch_size = batch_size
1703-
self.queries: List[QUERY] = []
1703+
self.queries: List[QueryBuilder] = []
17041704

17051705
def _make_query(self) -> None:
17061706
table = self.model._meta.basetable

0 commit comments

Comments
 (0)