Update inference.py

inference.py

@@ -268,8 +268,8 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, c
                 if len(safe_frame) == 0:
                     continue
 
-                has_next_batch = batch_i+1 < len(inference)
-                has_prev_batch = batch_i-1 >= 0
+                next_batch = inference[batch_i + 1] if batch_i+1 < len(inference) else None
+                prev_batch = inference[batch_i - 1] if batch_i-1 >= 0 else None
 
                 
                 for dt in range(-boost_range, boost_range+1):
@@ -278,11 +278,9 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, c
                     temp_frame = None
                     if idx >= 0 and idx < len(infer):
                         temp_frame = infer[idx]
-                    elif idx < 0 and has_prev_batch:
-                        prev_batch = inference[batch_i - 1]
+                    elif idx < 0 and prev_batch is not None and -idx >= len(prev_batch):
                         temp_frame = prev_batch[idx]
-                    elif idx >= len(infer) and has_next_batch:
-                        next_batch = inference[batch_i + 1]
+                    elif idx >= len(infer) and next_batch is not None and idx - len(infer) < len(next_batch):
                         temp_frame = next_batch[idx - len(infer)]
                     
                     if temp_frame is not None: