From 42909a94ac95d6bccef882c8af66aff925e625ff Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Fri, 13 Jan 2023 20:34:18 -0800 Subject: [PATCH] No need for np.repeat in normalize_along_axis --- manimlib/utils/space_ops.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/manimlib/utils/space_ops.py b/manimlib/utils/space_ops.py index 737966ad..a6bd3c71 100644 --- a/manimlib/utils/space_ops.py +++ b/manimlib/utils/space_ops.py @@ -193,8 +193,7 @@ def normalize_along_axis( ) -> np.ndarray: norms = np.sqrt((array * array).sum(axis)) norms[norms == 0] = 1 - buffed_norms = np.repeat(norms, array.shape[axis]).reshape(array.shape) - return array / buffed_norms + return (array.T / norms).T def get_unit_normal(