Skip to content

Commit ba3dbac

Browse files
committed
Add a DataArray accessor
1 parent 8212d04 commit ba3dbac

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

cubed_xarray/cubedmanager.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,31 @@ def store(
214214
)
215215

216216

217+
@xr.register_dataarray_accessor("cubed")
218+
class DataArrayAccessor:
219+
def __init__(self, da):
220+
self.da = da
221+
222+
def visualize(
223+
self,
224+
filename="cubed",
225+
format=None,
226+
optimize_graph=True,
227+
optimize_function=None,
228+
show_hidden=False,
229+
):
230+
import cubed
231+
232+
return cubed.visualize(
233+
self.da.data,
234+
filename=filename,
235+
format=format,
236+
optimize_graph=optimize_graph,
237+
optimize_function=optimize_function,
238+
show_hidden=show_hidden,
239+
)
240+
241+
217242
@xr.register_dataset_accessor("cubed")
218243
class DatasetAccessor:
219244
def __init__(self, ds):

0 commit comments

Comments
 (0)