Skip to content

Commit 5a9342a

Browse files
committed
allow to manage empty dataset for flatten_lin_matrix
1 parent 159642e commit 5a9342a

File tree

2 files changed

+7
-12
lines changed

2 files changed

+7
-12
lines changed

src/py_eddy_tracker/dataset/grid.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -307,18 +307,6 @@ def __init__(
307307
self.load_general_features()
308308
self.load()
309309

310-
def get_slice(self, **kwargs):
311-
"""
312-
Get slice for each axes specified
313-
314-
:param dict kwargs:
315-
:return: A dict of slice
316-
:rtype: dict[slice]
317-
"""
318-
for k, v in kwargs.items():
319-
with Dataset(self.filename) as h:
320-
print(k, v)
321-
322310
@property
323311
def is_centered(self):
324312
"""Give True if pixel is described with its center's position or

src/py_eddy_tracker/generic.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,14 @@ def flatten_line_matrix(l_matrix):
278278
"""
279279
nb_line, sampling = l_matrix.shape
280280
final_size = (nb_line - 1) + nb_line * sampling
281+
empty_dataset = False
282+
if final_size < 1:
283+
empty_dataset = True
284+
final_size = 1
281285
out = empty(final_size, dtype=l_matrix.dtype)
286+
if empty_dataset:
287+
out[:] = nan
288+
return out
282289
inc = 0
283290
for i in range(nb_line):
284291
for j in range(sampling):

0 commit comments

Comments
 (0)