From 85edb39e469847f29dd6ba63027ca6c0f57078e5 Mon Sep 17 00:00:00 2001 From: Andrew White Date: Mon, 27 Jul 2026 20:47:41 -0500 Subject: [PATCH] fix: Regr3D_ScaleInv.get_all_pts3d drops **kw from base --- dust3r/losses.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dust3r/losses.py b/dust3r/losses.py index 4f8febf..961a635 100644 --- a/dust3r/losses.py +++ b/dust3r/losses.py @@ -268,9 +268,9 @@ class Regr3D_ScaleInv (Regr3D): if gt_scale == True: enforce the prediction to take the same scale than GT """ - def get_all_pts3d(self, gt1, gt2, pred1, pred2): + def get_all_pts3d(self, gt1, gt2, pred1, pred2, **kw): # compute depth-normalized points - gt_pts1, gt_pts2, pred_pts1, pred_pts2, mask1, mask2, monitoring = super().get_all_pts3d(gt1, gt2, pred1, pred2) + gt_pts1, gt_pts2, pred_pts1, pred_pts2, mask1, mask2, monitoring = super().get_all_pts3d(gt1, gt2, pred1, pred2, **kw) # measure scene scale _, gt_scale = get_joint_pointcloud_center_scale(gt_pts1, gt_pts2, mask1, mask2)