small

Author: FedeClaudi

proj/__init__.py

@@ -1,3 +1,11 @@
+# Set up logging
+import logging
+
+for module in ["matplotlib", "pandas", "numpy"]:
+    requests_logger = logging.getLogger("matplotlib")
+    requests_logger.setLevel(logging.ERROR)
+
+# import stuff
 from proj.model.config import Config
 from proj.model.model import Model
 from proj.model.model_polar import ModelPolar

proj/environment/manager.py

@@ -30,7 +30,7 @@ def __init__(self, model, winstor=False):
         self.simstart = timestamp()
 
         self.model = model
-        self.exp_name = f'{model.trajectory["name"]}_{timestamp()}_{np.random.randint(low=0, high=10000)}'
+        self.exp_name = f'{model.SIMULATION_NAME}_{model.trajectory["name"]}_{timestamp()}_{np.random.randint(low=0, high=10000)}'
 
         # get main folder
         if winstor:

proj/model/config.py

@@ -2,6 +2,7 @@
 
 
 class Config:
+    SIMULATION_NAME = ""
     USE_FAST = True  # if true use cumba's methods
     SPAWN_TYPE = "trajectory"
 

workspace.py

@@ -1,5 +1,21 @@
-from proj.utils.dropbox import DropBoxUtils
+# %%
+from proj.utils.misc import load_results_from_folder
 
-dbx = DropBoxUtils()
+fld = "/Users/federicoclaudi/Dropbox (UCL - SWC)/Rotation_vte/Locomotion/control/tracking_200923_141709_1491_good_example/results"
 
-dbx.upload_file("winstor.py", "winstor.py")
+# %%
+config, trajectory, history, cost_history = load_results_from_folder(fld)
+
+# %%
+import matplotlib.pyplot as plt
+
+plt.plot(cost_history["x"])
+
+# %%
+plt.plot(history["tau_r"])
+plt.plot(history["tau_l"])
+
+# %%
+plt.plot(history["v"])
+
+# %%