From 1c3018812297a0c306e98dcbeb8c69b8d5cb3b13 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Sat, 6 Jun 2026 16:25:33 +0200
Subject: [PATCH] feat(checker): record type judgements

---
 midas/checker/checker.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/midas/checker/checker.py b/midas/checker/checker.py
index a96b472..8ca9b33 100644
--- a/midas/checker/checker.py
+++ b/midas/checker/checker.py
@@ -48,6 +48,7 @@ class Checker(
         self.env: Environment = self.global_env
         self.locals: dict[p.Expr, int] = locals
         self.diagnostics: list[Diagnostic] = []
+        self.judgements: list[tuple[p.Expr, Type]] = []
 
     def diagnostic(self, type: DiagnosticType, location: Location, message: str):
         self.diagnostics.append(
@@ -89,7 +90,9 @@ class Checker(
         Returns:
             Type: the type of the given expression
         """
-        return expr.accept(self)
+        type: Type = expr.accept(self)
+        self.judgements.append((expr, type))
+        return type
 
     def process_block(self, block: list[p.Stmt], env: Environment) -> bool:
         """Evaluate a sequence of statements