patch import of milestone types into test helpers

regro · Mar 24, 2024 · 2747988 · 2747988
1 parent f96f5a1
commit 2747988
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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 = [