code review part 2

mila-iqia · Oct 11, 2024 · 977f99b · 977f99b
1 parent c6f8f96
commit 977f99b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crystal_diffusion/models/egnn_utils.py b/crystal_diffusion/models/egnn_utils.py
@@ -3,7 +3,8 @@
 import torch
 
 from crystal_diffusion.models.mace_utils import get_adj_matrix
-from crystal_diffusion.utils.basis_transformations import get_positions_from_coordinates
+from crystal_diffusion.utils.basis_transformations import \
+    get_positions_from_coordinates
 
 
 def unsorted_segment_sum(data: torch.Tensor, segment_ids: torch.Tensor, num_segments: int) -> torch.Tensor: