From 0bebd680b5f233bf354fd6b33cae643ca2f89628 Mon Sep 17 00:00:00 2001
From: Michele Alberti <michele.alberti@cea.fr>
Date: Fri, 14 Apr 2023 16:43:57 +0200
Subject: [PATCH] [trans] Fix utility count_nn_classifiers to count only
different classifier symbols.
---
src/transformations/utils.ml | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/transformations/utils.ml b/src/transformations/utils.ml
index a2279ec..26966f6 100644
--- a/src/transformations/utils.ml
+++ b/src/transformations/utils.ml
@@ -42,10 +42,14 @@ let count_nn_classifiers =
| Term.Tapp (ls, _) -> (
match Language.lookup_nn_classifier ls with
| None -> acc
- | Some _ -> acc + 1)
+ | Some _ -> Term.Sls.add ls acc)
| _ -> acc
in
- Trans.fold_decl (fun decl acc -> Decl.decl_fold aux acc decl) 0
+ Trans.bind
+ (Trans.fold_decl
+ (fun decl acc -> Decl.decl_fold aux acc decl)
+ Term.Sls.empty)
+ (fun s -> Trans.return (Term.Sls.cardinal s))
let get_input_variables =
let add i acc = function
--
GitLab