From 692e23bced59ad204c5aff6a6773cc1578d89e51 Mon Sep 17 00:00:00 2001 From: Bruno Rousseau Date: Sat, 2 Mar 2024 05:18:53 -0500 Subject: [PATCH] Fix comment text. --- crystal_diffusion/score/wrapped_gaussian_score.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/crystal_diffusion/score/wrapped_gaussian_score.py b/crystal_diffusion/score/wrapped_gaussian_score.py index 6228afe0..44e4ea13 100644 --- a/crystal_diffusion/score/wrapped_gaussian_score.py +++ b/crystal_diffusion/score/wrapped_gaussian_score.py @@ -89,9 +89,9 @@ def get_sigma_normalized_score( # The dimension of list_k is [2 kmax + 1]. list_k = torch.arange(-kmax, kmax + 1) - # Initialize a results array, and view it as a column. - # Since "column_view" is a view on "sigma_normalized_scores", sigma_normalized_scores is updatedo - # when we assign in column_view. + # Initialize a results array, and view it as a flat list. + # Since "flat_view" is a view on "sigma_normalized_scores", sigma_normalized_scores is updated + # when we assign values in flat_view (both tensors share the same underlying data structure). sigma_normalized_scores = torch.zeros_like(relative_positions) flat_view = sigma_normalized_scores.view(total_number_of_elements)