Skip to content

Commit 0cb05b8

Browse files
Pathways-on-Cloud Teamcopybara-github
authored andcommitted
rm dead code
PiperOrigin-RevId: 918633772
1 parent 40a84b4 commit 0cb05b8

1 file changed

Lines changed: 0 additions & 28 deletions

File tree

pathwaysutils/persistence/helper.py

Lines changed: 0 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -251,34 +251,6 @@ def write_arrays(
251251
return bulk_write_future
252252

253253

254-
def read_one_array(
255-
location: str,
256-
name: str,
257-
dtype: np.dtype,
258-
shape: Sequence[int],
259-
shardings: jax.sharding.Sharding,
260-
devices: Sequence[jax.Device] | np.ndarray,
261-
timeout: datetime.timedelta,
262-
) -> jax.Array:
263-
"""Creates the read array plugin program string, compiles it to an executable, calls it and returns the result."""
264-
read_request = get_read_request(
265-
location,
266-
name,
267-
dtype,
268-
shape,
269-
shardings,
270-
devices,
271-
timeout,
272-
)
273-
read_executable = plugin_executable.PluginExecutable(read_request)
274-
out_aval = core.ShapedArray(shape, dtype)
275-
read_array, read_future = read_executable.call(
276-
out_shardings=[shardings], out_avals=[out_aval]
277-
)
278-
read_future.result()
279-
return read_array[0]
280-
281-
282254
def read_arrays(
283255
location: str,
284256
names: Sequence[str],

0 commit comments

Comments
 (0)