Fix type annotation

bhky · Dec 1, 2023 · 49c1100 · 49c1100
1 parent 78fb345
commit 49c1100
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opennsfw2/_inference.py b/opennsfw2/_inference.py
@@ -62,7 +62,7 @@ def _predict_images_in_batches(
     for i in range(0, len(images), batch_size):
         batch = np.array(images[i: i + batch_size])
         prediction_batches.append(model_(batch))
-    predictions = np.concatenate(prediction_batches, axis=0)
+    predictions: NDFloat32Array = np.concatenate(prediction_batches, axis=0)
     return predictions