lint format fix

adaptive-intelligent-robotics · Jan 8, 2025 · 0a8b8df · 0a8b8df
1 parent ed7e79f
commit 0a8b8df
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/qdax/core/containers/unstructured_repertoire.py b/qdax/core/containers/unstructured_repertoire.py
@@ -128,11 +128,12 @@ def intra_batch_comp(
     true_positions = jnp.nonzero(eq_mask, size=eq_mask.size, fill_value=-1)[0]
     tie_indices = jnp.take(knn_relevant_indices, true_positions, mode='clip')
 
-    # Instead of keeping the minimal indexed individual, we keep the first tied individual
-    # as encountered in knn_relevant_indices ordering.
+    # Instead of keeping the minimal indexed individual, we keep the first tied
+    # individual as encountered in knn_relevant_indices ordering.
     keep_index = tie_indices[0]  # First individual in the tie list
 
-    # If there's more than one individual tied, discard the current one if it's not the "keep_index"
+    # If there's more than one individual tied,
+    # discard the current one if it's not the "keep_index"
     tie_discard = jnp.logical_and((tie_indices.size > 1), (current_index != keep_index))
 
     discard_indiv = jnp.logical_or(discard_indiv, tie_discard)