Hey @sandy, based on the example, I realized that the partition_mapping is not working as I had it in mind: the test case shows, that both assets/ops are only called with partition_key 2; the extended mapping is only used in the IOManager. I would have thought that a job with the two provided assets should lead to three asset materialization events (upstream asset with partitions 1 and 2), otherwise, the IOManager tries to load an unmaterialized asset partition.
Is this the intended behavior and if so, how would one create a job with both assets, that ensures all required partitions are materialized?