From 274798883f79ff5db3ac36e45a74241a9530ccf7 Mon Sep 17 00:00:00 2001
From: Simon Billinge <simon.billinge@gmail.com>
Date: Sun, 24 Mar 2024 16:31:44 -0400
Subject: [PATCH] patch import of milestone types into test helpers

---
 tests/test_helpers.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
index 6fedbb85c..3f8d8246b 100644
--- a/tests/test_helpers.py
+++ b/tests/test_helpers.py
@@ -6,7 +6,9 @@
 import requests_mock
 
 from regolith.main import main
-from regolith.schemas import MILESTONE_TYPES
+from regolith.schemas import alloweds
+
+MILESTONE_TYPES = alloweds.get("MILESTONE_TYPES")
 
 dash = "-"
 helper_map = [