Add import statements for Message and MessagesList

sparkles/core.py

@@ -20,6 +20,7 @@
 
 from sparkles import checks
 from sparkles.aca_checks_table import ACAChecksTable
+from sparkles.messages import Message, MessagesList
 from sparkles.roll_optimize import RollOptimizeMixin
 
 CACHE = {}
@@ -556,7 +557,7 @@ def get_summary_text(acas):
         )
 
         # Warnings
-        for category in reversed(checks.MessagesList.categories):
+        for category in reversed(MessagesList.categories):
             msgs = aca.messages == category
             if msgs:
                 text = stylize(f" {category.capitalize()}: {len(msgs)}", category)
@@ -773,7 +774,7 @@ def make_roll_options_report(self):
         msgs_summaries = []
         for aca in acas:
             texts = []
-            for category in reversed(checks.MessagesList.categories):
+            for category in reversed(MessagesList.categories):
                 msgs = aca.messages == category
                 if msgs:
                     text = stylize(f"{category.capitalize()}: {len(msgs)}", category)
@@ -1034,7 +1035,7 @@ def from_ocat(cls, obsid, t_ccd=-5, man_angle=5, date=None, roll=None, **kwargs)
 
 def check_catalog(acar: ACAChecksTable) -> None:
     """Perform all star catalog checks."""
-    msgs: list[checks.Message] = []
+    msgs: list[Message] = []
     for entry in acar:
         entry_type = entry["type"]
         is_guide = entry_type in ("BOT", "GUI")