File tree Expand file tree Collapse file tree
pathwaysutils/persistence Expand file tree Collapse file tree Original file line number Diff line number Diff 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-
282254def read_arrays (
283255 location : str ,
284256 names : Sequence [str ],
You can’t perform that action at this time.
0 commit comments