From 8b7994869c98119e50e8d19f857baccad7194628 Mon Sep 17 00:00:00 2001
From: Simmo Saan <simmo.saan@gmail.com>
Date: Tue, 21 Nov 2023 11:21:46 +0200
Subject: [PATCH] Fix invariant_set elements schema in YAML witnesses

---
 src/witness/yamlWitnessType.ml | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/witness/yamlWitnessType.ml b/src/witness/yamlWitnessType.ml
index f9bcf3235f..de9fa151d8 100644
--- a/src/witness/yamlWitnessType.ml
+++ b/src/witness/yamlWitnessType.ml
@@ -311,13 +311,16 @@ struct
     }
 
     let to_yaml {invariant_type} =
-      `O ([
-          ("type", `String (InvariantType.invariant_type invariant_type));
-        ] @ InvariantType.to_yaml' invariant_type)
+      `O [
+        ("invariant", `O ([
+             ("type", `String (InvariantType.invariant_type invariant_type));
+           ] @ InvariantType.to_yaml' invariant_type)
+        )
+      ]
 
     let of_yaml y =
       let open GobYaml in
-      let+ invariant_type = y |> InvariantType.of_yaml in
+      let+ invariant_type = y |> find "invariant" >>= InvariantType.of_yaml in
       {invariant_type}
   end