remove additional np.object references

twosixlabs · Nov 6, 2023 · 385c183 · 385c183
1 parent 042c062
commit 385c183
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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: