Improve style.

Adibvafa · Sep 20, 2024 · e6c3ed5 · e6c3ed5
1 parent dfd11f6
commit e6c3ed5
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/CodonTransformer/CodonPrediction.py b/CodonTransformer/CodonPrediction.py
@@ -193,7 +193,6 @@ def predict_dna_sequence(
         if deterministic:
             predicted_indices = logits.argmax(dim=-1).squeeze().tolist()
         else:
-            # Use the standalone non-deterministic sampling function
             predicted_indices = sample_non_deterministic(
                 logits=logits, temperature=temperature, top_p=top_p
             )