Merge pull request #325 from os-climate/vault-mypy

Update to Pydantic 2.x
os-climate · Nov 25, 2023 · 04c0d9f · 04c0d9f
2 parents af7b6c7 + b7bf915
commit 04c0d9f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/ITR/data/vault_providers.py b/src/ITR/data/vault_providers.py
@@ -464,9 +464,9 @@ def __init__(
         )
         df = pd.DataFrame()
         for scope in ["AnyScope"]:
-            if production_benchmarks.dict()[scope] is None:
+            if production_benchmarks.model_dump()[scope] is None:
                 continue
-            for benchmark in production_benchmarks.dict()[scope]["benchmarks"]:
+            for benchmark in production_benchmarks.model_dump()[scope]["benchmarks"]:
                 bdf = pd.DataFrame.from_dict(
                     {
                         r["year"]: [
@@ -519,9 +519,9 @@ def __init__(
         self.projection_controls = projection_controls
         df = pd.DataFrame()
         for scope in EScope.get_scopes():
-            if EI_benchmarks.dict()[scope] is None:
+            if EI_benchmarks.model_dump()[scope] is None:
                 continue
-            for benchmark in EI_benchmarks.dict()[scope]["benchmarks"]:
+            for benchmark in EI_benchmarks.model_dump()[scope]["benchmarks"]:
                 benchmark_df = pd.DataFrame.from_dict(
                     {
                         r["year"]: [