diff --git a/dbt_common/clients/agate_helper.py b/dbt_common/clients/agate_helper.py index 614d8815..719d49c7 100644 --- a/dbt_common/clients/agate_helper.py +++ b/dbt_common/clients/agate_helper.py @@ -1,6 +1,6 @@ from codecs import BOM_UTF8 -import agate +import agate # type: ignore import datetime import isodate import json diff --git a/dbt_common/events/functions.py b/dbt_common/events/functions.py index 115d8553..a21fd317 100644 --- a/dbt_common/events/functions.py +++ b/dbt_common/events/functions.py @@ -94,7 +94,7 @@ def msg_to_dict(msg: EventMsg) -> dict: msg_dict = dict() try: msg_dict = MessageToDict( - msg, + msg.data, preserving_proto_field_name=True, including_default_value_fields=True, # type: ignore ) diff --git a/dbt_common/utils/encoding.py b/dbt_common/utils/encoding.py index ab19b894..96acb9c8 100644 --- a/dbt_common/utils/encoding.py +++ b/dbt_common/utils/encoding.py @@ -9,7 +9,7 @@ DECIMALS: Tuple[Type[Any], ...] try: - import cdecimal # typing: ignore + import cdecimal # type: ignore except ImportError: DECIMALS = (decimal.Decimal,) else: diff --git a/pyproject.toml b/pyproject.toml index a8c49173..b74190b5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -70,6 +70,7 @@ dependencies = [ "pytest~=7.3", "types-Jinja2~=2.11", "types-jsonschema~=4.17", + "types-protobuf~=4.24.0", "types-python-dateutil~=2.8", "types-PyYAML~=6.0", "types-requests<2.31.0" # types-requests 2.31.0.8 requires urllib3>=2, but we pin urllib3 ~= 1.0 because of openssl requirement for requests diff --git a/third-party-stubs/jinja2/__init__.py b/third-party-stubs/jinja2/__init__.py deleted file mode 100644 index e69de29b..00000000