Skip to content

Commit 98282e0

Browse files
committed
MNT: add required keys to steps and image
1 parent 0eac35e commit 98282e0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

data_prototype/wrappers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,7 @@ def _update_wrapped(self, data):
217217

218218
class ImageWrapper(ProxyWrapper):
219219
_wrapped_class = _AxesImage
220+
required_keys = {"xextent", "yextent", "image"}
220221

221222
def __init__(self, data: DataContainer, nus=None, /, cmap=None, norm=None, **kwargs):
222223
nus = dict(nus or {})
@@ -247,6 +248,7 @@ def _update_wrapped(self, data):
247248
class StepWrapper(ProxyWrapper):
248249
_wrapped_class = _StepPatch
249250
_privtized_methods = () # ("set_data", "get_data")
251+
required_keys = {"edges", "density"}
250252

251253
def __init__(self, data: DataContainer, nus=None, /, **kwargs):
252254
super().__init__(data, nus)

0 commit comments

Comments
 (0)