From 7441db19749fb7fb9383b6361dfbd99547e53486 Mon Sep 17 00:00:00 2001
From: Alasdair
Date: Tue, 29 Sep 2020 16:23:40 +0100
Subject: Refactor: Change AST type from a union to a struct

---
 src/optimize.ml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/optimize.ml')

diff --git a/src/optimize.ml b/src/optimize.ml
index 8c456a1b..fccf0f3a 100644
--- a/src/optimize.ml
+++ b/src/optimize.ml
@@ -65,7 +65,7 @@ let split_at_function id defs =
   | Some (pre_defs, def, post_defs) ->
      Some (List.rev pre_defs, def, post_defs)
 
-let recheck (Defs defs) =
+let recheck ({ defs; _} as ast) =
   let defs = Type_check.check_with_envs Type_check.initial_env defs in
 
   let rec find_optimizations = function
@@ -112,4 +112,4 @@ let recheck (Defs defs) =
     | [] -> []
   in
 
-  Defs (find_optimizations defs)
+  { ast with defs = find_optimizations defs }
-- 
cgit v1.2.3