fix: Fix function args.

SysCV · Oct 15, 2024 · 53f2242 · 53f2242
1 parent d95fb72
commit 53f2242
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vis4d/eval/common/depth.py b/vis4d/eval/common/depth.py
@@ -100,8 +100,8 @@ def process_batch(  # type: ignore # pylint: disable=arguments-differ
         """Process a batch of data.
 
         Args:
-            prediction (np.array): Prediction optical flow, in shape (H, W, 2).
-            groundtruth (np.array): Target optical flow, in shape (H, W, 2).
+            prediction (np.array): Prediction optical flow, in shape (B, H, W).
+            groundtruth (np.array): Target optical flow, in shape (B, H, W).
         """
         preds = (
             array_to_numpy(prediction, n_dims=None, dtype=np.float32)