From 6ee926c68bd25ad155e7085b552e8d9a3109de92 Mon Sep 17 00:00:00 2001
From: Tim Jammer <tim.jammer@tu-darmstadt.de>
Date: Mon, 29 Apr 2024 15:02:31 +0200
Subject: [PATCH] fix to use new level

---
 scripts/errors/coll/ParamMatchingType.py | 4 ++--
 scripts/main.py                          | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/errors/coll/ParamMatchingType.py b/scripts/errors/coll/ParamMatchingType.py
index 9601ea314..f7bc41310 100644
--- a/scripts/errors/coll/ParamMatchingType.py
+++ b/scripts/errors/coll/ParamMatchingType.py
@@ -184,12 +184,12 @@ class InvalidComErrorColl(ErrorGenerator):
         for f in self.functions_to_use:
             for comm in predefined_comms + comm_creators + intercomms:
                 for i, t1 in enumerate(types):
-                    if generate_level == REAL_WORLD_TEST_LEVEL and not is_combination_important(real_world_score_table,
+                    if generate_level in REAL_WORLD_FILTERING_LEVELS and not is_combination_important(real_world_score_table,
                                                                                                 f, datatype=t1.lower(),
                                                                                                 communicator=comm):
                         continue
                     for t2 in types[i:]:
-                        if generate_level == REAL_WORLD_TEST_LEVEL and not is_combination_important(
+                        if generate_level in REAL_WORLD_FILTERING_LEVELS and not is_combination_important(
                                 real_world_score_table, f, datatype=t2.lower(), communicator=comm):
                             continue
                         if is_combination_compatible(t1, t2, f):
diff --git a/scripts/main.py b/scripts/main.py
index 48d73a56f..2844c20c3 100644
--- a/scripts/main.py
+++ b/scripts/main.py
@@ -31,7 +31,7 @@ if __name__ == "__main__":
     gencodes_dir = ARGS.outpath
 
     score_table = None
-    if ARGS.level == REAL_WORLD_TEST_LEVEL:
+    if ARGS.level in REAL_WORLD_FILTERING_LEVELS:
         print("Read real world dataset and build scoring table")
         score_table = get_scoring_table(pd.read_csv(ARGS.real_world_data, header=0, low_memory=False), c_only=True)
 
-- 
GitLab