Skip to content

Commit ea5082b

Browse files
Donglai Weiclaude
andcommitted
Fix decode-only: return TTA suffix so pipeline runs decoding on loaded predictions
saved_prediction_path was returning a non-TTA suffix, causing the pipeline to treat it as a final prediction and skip decoding entirely. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 881ebe8 commit ea5082b

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

connectomics/training/lightning/model.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -604,7 +604,9 @@ def _load_cached_predictions(
604604
pred = read_volume(str(pred_file), dataset="main")
605605
if pred.ndim < 4:
606606
pred = pred[np.newaxis, ...]
607-
return pred, True, cache_suffix
607+
# Return a TTA-like suffix so the pipeline treats this as
608+
# intermediate predictions (runs decoding), not final.
609+
return pred, True, "_tta_x1_prediction.h5"
608610
else:
609611
raise FileNotFoundError(
610612
f"inference.saved_prediction_path not found: {pred_file}"

0 commit comments

Comments
 (0)