dynamic partition names

2025-08-02 10:49:56 +02:00
parent 2a032cf25d
commit b5f7952c07
3 changed files with 7 additions and 4 deletions
--- a/shared/src/shared/utils.py
+++ b/shared/src/shared/utils.py
@@ -21,7 +21,7 @@ def get_dimension_names(
    """
    partition_definition = context.asset_partitions_def_for_input(input_name)
    if isinstance(partition_definition, dg.MultiPartitionsDefinition):
-        return [x.name for x in partition_definition.partitions_defs]
+        return partition_definition.partition_dimension_names
    raise NotImplementedError("Only MultiPartitionsDefinition is supported.")