diff --git a/src/sentry/db/models/query.py b/src/sentry/db/models/query.py index f763f40c5c392b..0c7882fea18a51 100644 --- a/src/sentry/db/models/query.py +++ b/src/sentry/db/models/query.py @@ -62,7 +62,7 @@ def _resolve(instance: Model, node: BaseExpression | F) -> BaseExpression: return runner -def _get_field(model: type[BaseModel], key: str) -> Field[object, object]: +def _get_field(model: type[Model], key: str) -> Field[object, object]: field = model._meta.get_field(key) if not isinstance(field, Field): raise TypeError(f"expected Field for {key}, got ({field})") @@ -171,8 +171,8 @@ def update_or_create( def create_or_update( - model: type[BaseModel], using: str | None = None, **kwargs: Any -) -> tuple[int, Literal[False]] | tuple[BaseModel, Literal[True]]: + model: type[Model], using: str | None = None, **kwargs: Any +) -> tuple[int, Literal[False]] | tuple[Model, Literal[True]]: """ Similar to get_or_create, either updates a row or creates it.