context.partition_key

class CustomIOManager(ConfigurableIOManager):
    def handle_output(self, context: OutputContext, obj: Any) -> None:
        print("my partition key", context.partition_key)
        return None

    def load_input(self, context: InputContext) -> Any:
        ...

@io_manager
def custom_io_manager():
    return CustomIOManager()


@static_partitioned_config(partition_keys=CONTINENTS)
def continent_config(partition_key: str):
    return {"ops": {"continent_op": {"config": {"continent_name": partition_key}}}}


class ContinentOpConfig(Config):
    continent_name: str


@op
def continent_op(context, config: ContinentOpConfig):
    <http://context.log.info|context.log.info>(config.continent_name)


@job(config=continent_config, resource_defs={"io_manager": custom_io_manager})
def continent_job():
    continent_op()