I m trying to set up a custom IO manager that handles partit

Question

I m trying to set up a custom IO manager that handles partition keys and I m a bit confused about how to implement `load input` properly In the case where a partitioned asset materialization is loading a non partitioned asset it seems like `InputContext partition key` is still set to the partition key of the materialization Does that mean I cannot deduce from the context passed to `load input` whether the upstream asset is partitioned or not

Erwin Kuhn · Answer

After looking at `UPathIOManager` which is the base class for the `fs io manager` it seems like looking at `context has asset partitions` and `context asset partition keys` is the way to go <https github com dagster io dagster blob master python modules dagster dagster core storage upath io manager py L134 L205>