refactor: Update robot_env.py to include current working directory in…

… filename for recordings
ll7 · Sep 25, 2024 · bb6c09d · bb6c09d
1 parent 4e4e179
commit bb6c09d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/robot_sf/gym_env/robot_env.py b/robot_sf/gym_env/robot_env.py
@@ -259,7 +259,9 @@ def save_recording(self, filename: str = None):
         """
         if filename is None:
             now = datetime.datetime.now()
-            filename = f'recordings/{now.strftime("%Y-%m-%d_%H-%M-%S")}.pkl'
+            # get current working directory
+            cwd = os.getcwd()
+            filename = f'{cwd}/recordings/{now.strftime("%Y-%m-%d_%H-%M-%S")}.pkl'
 
         # only save if there are recorded states
         if len(self.recorded_states) == 0: