fix linting, still some pending ignored errors

dbt-labs · Jan 12, 2024 · 457e0ba · 457e0ba
1 parent 4ddbdcd
commit 457e0ba
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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