From 250a2b88b278a878d2cf6743b5b7c665afee2638 Mon Sep 17 00:00:00 2001
From: Daniel Smith <56164590+DanielRyanSmith@users.noreply.github.com>
Date: Sun, 22 Oct 2023 13:56:07 -0700
Subject: [PATCH] Update maintenance_scripts.py

---
 internals/maintenance_scripts.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/internals/maintenance_scripts.py b/internals/maintenance_scripts.py
index 2651d98ce575..6294db3ab222 100644
--- a/internals/maintenance_scripts.py
+++ b/internals/maintenance_scripts.py
@@ -370,6 +370,9 @@ def clear_extension_requests(self, ot_stage: Stage, trial_data: dict) -> int:
       return 0
     extension_stages_to_update = []
     for extension_stage in extension_stages:
+      # skip the stage if it doesn't have an end milestone explicitly defined.
+      if extension_stage.milestones is None:
+        continue
       extension_end = extension_stage.milestones.desktop_last
       # If the end milestone of the trial is equal or greater than the
       # requested end milestone on the extension stage, we can assume the