Cleans up a trivial unnecessary logic.

PiperOrigin-RevId: 620433429
google · Mar 30, 2024 · 198029a · 198029a
1 parent ab88791
commit 198029a
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/praxis/sample_decode.py b/praxis/sample_decode.py
@@ -1281,10 +1281,6 @@ def put(target, key, val):
     result_callback.done_fn()
 
   if optimize_eos:
-    if fprop_for_prefix:
-      decode_length_shift = max_prefix_len
-    else:
-      decode_length_shift = 0
     result = decoder_utils.collect_results_to_optimize_eos(
         result, decode_length_shift=max_prefix_len
     )