towards detecting new deal

2025-07-26 19:09:06 +02:00
parent 8d06b236b7
commit 66bcb3e2d3
4 changed files with 61 additions and 25 deletions
--- a/shared/src/shared/utils.py
+++ b/shared/src/shared/utils.py
@@ -1,4 +1,4 @@
-from collections.abc import Generator
+from collections.abc import Iterator
 from typing import Mapping

 import polars as pl
@@ -87,7 +87,7 @@ def parse_partition_keys(

 def load_partitions(
    context: dg.AssetExecutionContext, asset_key: dg.AssetKey, partitions: set[str]
-) -> Generator[pl.DataFrame, None, None]:
+) -> Iterator[pl.DataFrame]:
    """
    Load data from an asset for the specified partitions.