From 9afa17680698d9d44cb343253fad7a5f0c44b9a7 Mon Sep 17 00:00:00 2001 From: Marton Havasi Date: Tue, 10 Dec 2024 19:01:19 +0000 Subject: [PATCH] grammar nit --- flow_matching/utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flow_matching/utils/utils.py b/flow_matching/utils/utils.py index 0b170db..23dd1db 100644 --- a/flow_matching/utils/utils.py +++ b/flow_matching/utils/utils.py @@ -50,7 +50,7 @@ def expand_tensor_like(input_tensor: Tensor, expand_to: Tensor) -> Tensor: Tensor: (B, ...). """ assert input_tensor.ndim == 1, "Input tensor must be a 1d vector." - assert input_tensor.shape[0] == expand_to.shape[0], f"The first (batch) dimensions must match. Got shape {input_tensor.shape} and {expand_to.shape}." + assert input_tensor.shape[0] == expand_to.shape[0], f"The first (batch) dimension must match. Got shape {input_tensor.shape} and {expand_to.shape}." dim_diff = expand_to.ndim - input_tensor.ndim