From 385c18357a285f2bfe59113d7f9b369646335b39 Mon Sep 17 00:00:00 2001 From: Sterling Suggs Date: Mon, 6 Nov 2023 08:21:15 -0500 Subject: [PATCH] remove additional np.object references --- armory/data/datasets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/armory/data/datasets.py b/armory/data/datasets.py index cec605402..5f6d96c26 100644 --- a/armory/data/datasets.py +++ b/armory/data/datasets.py @@ -828,7 +828,7 @@ def __init__(self, x_shape, sample_rate, input_type=np.int64): def canonical_audio_preprocess(context, batch): - if batch.dtype == np.object: + if batch.dtype == object: for x in batch: check_shapes(x.shape, context.x_shape) assert x.dtype == context.input_type @@ -1692,8 +1692,8 @@ def __init__(self, max_frames): self.max_frames = max_frames def __call__(self, batch): - if batch.dtype == np.object: - clipped_batch = np.empty_like(batch, dtype=np.object) + if batch.dtype == object: + clipped_batch = np.empty_like(batch, dtype=object) clipped_batch[:] = [x[: self.max_frames] for x in batch] return clipped_batch else: