From 8de73738c7939cdc22cab62b5cf7b008fde0458b Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 4 Apr 2024 04:59:29 +0000 Subject: [PATCH] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20form?= =?UTF-8?q?at=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/geometry/utils.py | 4 ++-- test/geometry/test_utils.py | 40 +++++++++++++++++------------------ 2 files changed, 22 insertions(+), 22 deletions(-) diff --git a/supervision/geometry/utils.py b/supervision/geometry/utils.py index 401f672e..48a19b2d 100644 --- a/supervision/geometry/utils.py +++ b/supervision/geometry/utils.py @@ -36,7 +36,7 @@ def get_polygon_center(polygon: np.ndarray) -> Point: shifted_polygon = np.roll(polygon, 1, axis=0) points = (shifted_polygon + polygon) / 2 vectors = shifted_polygon - polygon - mass = np.sum(vectors ** 2, axis=1) ** 0.5 + mass = np.sum(vectors**2, axis=1) ** 0.5 center = ((mass @ points) / np.sum(mass)).round() - + return Point(x=center[0], y=center[1]) diff --git a/test/geometry/test_utils.py b/test/geometry/test_utils.py index bb6ec60b..46963716 100644 --- a/test/geometry/test_utils.py +++ b/test/geometry/test_utils.py @@ -7,32 +7,32 @@ from supervision.geometry.utils import get_polygon_center def generate_test_polygon(n: int) -> np.ndarray: """ - Generate a semicircle with a given number of points. - - Parameters: - n (int): amount of points in polygon + Generate a semicircle with a given number of points. - Returns: - Polygon: test polygon in the form of a semicircle. - - Examples: - ```python - from supervision.geometry.utils import get_polygon_center - import numpy as np - - test_polygon = generate_test_data(1000) - - get_polygon_center(test_polygon) - Point(x=500, y=1212) - ``` + Parameters: + n (int): amount of points in polygon + + Returns: + Polygon: test polygon in the form of a semicircle. + + Examples: + ```python + from supervision.geometry.utils import get_polygon_center + import numpy as np + + test_polygon = generate_test_data(1000) + + get_polygon_center(test_polygon) + Point(x=500, y=1212) + ``` """ r: int = n // 2 x_axis = np.linspace(0, 2 * r, n) - y_axis = (r ** 2 - (x_axis - r) ** 2) ** 0.5 + 2 * r + y_axis = (r**2 - (x_axis - r) ** 2) ** 0.5 + 2 * r polygon = np.array([x_axis, y_axis]).T - + return polygon - + @pytest.mark.parametrize( "polygon, expected_result",