From 83bfc696bceb9a642794c96875789a0e6becd5bd Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Mon, 26 Oct 2026 15:43:31 -0700 Subject: [PATCH] Use _device_assignment in sidechannel reshard request. PiperOrigin-RevId: 1007611830 --- pathwaysutils/experimental/reshard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pathwaysutils/experimental/reshard.py b/pathwaysutils/experimental/reshard.py index a027092..1f156ff 100644 --- a/pathwaysutils/experimental/reshard.py +++ b/pathwaysutils/experimental/reshard.py @@ -61,7 +61,7 @@ def ifrt_hlo_sharding( result = { "devices": { "device_ids": [ - device.id for device in sharding._addressable_device_assignment # pylint: disable=protected-access + device.id for device in sharding._device_assignment # pylint: disable=protected-access ] }, "xla_hlo_sharding": (