Fix typo in mask creation (#3)

althonos · Sep 15, 2024 · c04bb3d · c04bb3d
1 parent 11a8498
commit c04bb3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mini3di/encoder.py b/mini3di/encoder.py
@@ -230,7 +230,7 @@ def _create_nan_mask(
         """
         mask_ca = numpy.isnan(ca).max(axis=1)
         mask_n = numpy.isnan(n).max(axis=1)
-        mask_c = numpy.isnan(n).max(axis=1)
+        mask_c = numpy.isnan(c).max(axis=1)
         return (mask_ca | mask_n | mask_c).repeat(3).reshape(-1, 3)
 
     def encode_atoms(