Fix air_private_input_comparator.py

vm/src/tests/air_private_input_comparator.py

@@ -7,6 +7,12 @@
 filename1 = sys.argv[1]
 filename2 = sys.argv[2]
 
+def map_to_vec(batch):
+    if isinstance(batch, list):
+        return batch
+
+    return list(batch.values())
+
 with open(filename1, 'r') as cairo_lang_input_file, open(filename2, 'r') as cairo_vm_input_file:
     cairo_lang_input = json.load(cairo_lang_input_file)
     cairo_vm_input = json.load(cairo_vm_input_file)
@@ -18,6 +24,16 @@
     cairo_vm_input["trace_path"] = ""
     cairo_vm_input["memory_path"] = ""
 
+    # The trace_path & memory_path fields contain a different format for the modulo instance batches
+    for i, instance in enumerate(cairo_vm_input["add_mod"]["instances"]):
+        instance["batch"] = map_to_vec(instance["batch"])
+    for i, instance in enumerate(cairo_vm_input["mul_mod"]["instances"]):
+        instance["batch"] = map_to_vec(instance["batch"])
+    for i, instance in enumerate(cairo_lang_input["add_mod"]["instances"]):
+        instance["batch"] = map_to_vec(instance["batch"])
+    for i, instance in enumerate(cairo_lang_input["mul_mod"]["instances"]):
+        instance["batch"] = map_to_vec(instance["batch"])
+
     if cairo_lang_input == cairo_vm_input:
 
         print(f"Comparison succesful for {filename1} vs {filename2}")