From 1fffbeebd86485d1d9620136b76ce18ecbec8c34 Mon Sep 17 00:00:00 2001 From: Matthew Whitworth Date: Fri, 14 Jul 2023 12:11:37 +0100 Subject: [PATCH] ensure bbox_hat has right dimensions --- rrcf/rrcf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rrcf/rrcf.py b/rrcf/rrcf.py index 40b525b..3476739 100644 --- a/rrcf/rrcf.py +++ b/rrcf/rrcf.py @@ -1101,7 +1101,7 @@ def _insert_point_cut(self, point, bbox): (0, 0.9758881798109296) """ # Generate the bounding box - bbox_hat = np.empty(bbox.shape) + bbox_hat = np.empty((2, bbox.shape[1])) # Update the bounding box based on the internal point bbox_hat[0, :] = np.minimum(bbox[0, :], point) bbox_hat[-1, :] = np.maximum(bbox[-1, :], point)