From f81c04817046a59873560df34f3905e39271422a Mon Sep 17 00:00:00 2001
From: Antonio Andriella <aandriella@iri.upc.edu>
Date: Tue, 15 Sep 2020 11:31:06 +0200
Subject: [PATCH] add new variables from the BNs

---
 bn_variables.py | 26 ++++++++++----------------
 1 file changed, 10 insertions(+), 16 deletions(-)

diff --git a/bn_variables.py b/bn_variables.py
index af30a0d..3571ef6 100644
--- a/bn_variables.py
+++ b/bn_variables.py
@@ -1,19 +1,12 @@
 '''this module collect all the variables involved in the bayesian network and initialise them'''
 import enum
 
-class Robot_Assistance_Feedback(enum.Enum):
-    lev_0_no = 0
-    lev_1_no = 1
-    lev_2_no = 2
-    lev_3_no = 3
-    lev_4_no = 4
-    lev_0_yes = 0
-    lev_1_yes = 1
-    lev_2_yes = 2
-    lev_3_yes = 3
-    lev_4_yes = 4
-    name = "robot_assistance_feedback"
-    counter = 10
+class User_React_time(enum.Enum):
+    slow = 0
+    normal = 1
+    fast = 1
+    name = "user_react_time"
+    counter = 3
 
 class User_Capability(enum.Enum):
     very_mild = 0
@@ -56,12 +49,13 @@ class Robot_Assistance(enum.Enum):
     lev_2 = 2
     lev_3 = 3
     lev_4 = 4
+    lev_5 = 5
     name = "robot_assistance"
-    counter = 5
+    counter = 6
 
 class Robot_Feedback(enum.Enum):
-    yes = 0
-    no = 1
+    yes = 1
+    no = 0
     name = "robot_feedback"
     counter = 2
 
-- 
GitLab