From 1d701876be8f24d52c785194a7fa996a033be1cf Mon Sep 17 00:00:00 2001
From: Antonio Andriella <aandriella@iri.upc.edu>
Date: Thu, 24 Sep 2020 23:42:53 +0200
Subject: [PATCH] replace Robot with Agent

---
 cognitive_game_vars.py |  4 ++--
 episode.py             | 10 +++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/cognitive_game_vars.py b/cognitive_game_vars.py
index 3571ef6..4fa0f4b 100644
--- a/cognitive_game_vars.py
+++ b/cognitive_game_vars.py
@@ -43,7 +43,7 @@ class Attention(enum.Enum):
     name = "attention"
     counter = 3
 
-class Robot_Assistance(enum.Enum):
+class Agent_Assistance(enum.Enum):
     lev_0 = 0
     lev_1 = 1
     lev_2 = 2
@@ -53,7 +53,7 @@ class Robot_Assistance(enum.Enum):
     name = "robot_assistance"
     counter = 6
 
-class Robot_Feedback(enum.Enum):
+class Agent_Feedback(enum.Enum):
     yes = 1
     no = 0
     name = "robot_feedback"
diff --git a/episode.py b/episode.py
index cdf63e8..bcc420a 100644
--- a/episode.py
+++ b/episode.py
@@ -74,8 +74,8 @@ class Episode:
         '''
         It returns the episodes related to the saved file
         :param file:
-        :param episode: look at main.py
-        :param sol_per_pop: look at main.py
+        :param episode: look at simulation.py
+        :param sol_per_pop: look at simulation.py
         :return: a list of episodes
         '''
         print("LOADING...")
@@ -167,7 +167,7 @@ class Episode:
 
 def main():
     pass
-    # file_path = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/episodes.npy"
+    # file_path = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/episodes.npy"
     # ep = Episode()
     # episodes = ep.load_episodes(file_path)
     # initial_state = (1, 1, 0)
@@ -187,8 +187,8 @@ def main():
     #                              task_length, n_max_attempt, timeout=0, n_levels_assistance=6)
     # #
     # trans_matrix = ep.generate_statistics(env.states, env.action_space, episodes)
-    # path_trans_matrix_occ = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/trans_matrix_occ.npy"
-    # path_trans_matrix_prob = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/BN_GenerativeModel/results/1/trans_matrix_prob.npy"
+    # path_trans_matrix_occ = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/trans_matrix_occ.npy"
+    # path_trans_matrix_prob = "/home/aandriella/Documents/Codes/MY_FRAMEWORK/bn_generative_model/results/1/trans_matrix_prob.npy"
     # terminal_states = [env.point_to_index(state) for state in final_states]
     #
     #
-- 
GitLab