def METHOD_NAME(process_id):
    sharding = PositionalSharding(jax.devices())
    run_distributed_sharing_test(sharding=sharding, process_id=process_id)
    log.info("Passed positional sharding workflow test")