fix works asset

shared/src/shared/utils.py

@@ -1,8 +1,6 @@
 from collections.abc import Iterator
 from typing import Mapping
 
-import polars as pl
-
 import dagster as dg
 
 
@@ -87,7 +85,7 @@ def parse_partition_keys(
 
 def load_partitions(
     context: dg.AssetExecutionContext, asset_key: dg.AssetKey, partitions: set[str]
-) -> Iterator[pl.DataFrame]:
+) -> Iterator[object]:
     """
     Load data from an asset for the specified partitions.
 
@@ -99,8 +97,12 @@ def load_partitions(
     Yields:
         DataFrames for each partition specified.
     """
-    from definitions import definitions
 
-    loader = definitions.get_asset_value_loader(instance=context.instance)
+    loader = context.repository_def.get_asset_value_loader(instance=context.instance)
     for partition_key in partitions:
-        yield loader.load_asset_value(asset_key=asset_key, partition_key=partition_key)
+        if (
+            value := loader.load_asset_value(
+                asset_key=asset_key, partition_key=partition_key
+            )
+        ) is not None:
+            yield value