things kinda work; still struggling to get a noticeable difference be…

…tween ours and task-repeat tho...
Learning-and-Intelligent-Systems · Oct 10, 2023 · 9997c1b · 9997c1b
1 parent 350e0b7
commit 9997c1b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/predicators/envs/sticky_table.py b/predicators/envs/sticky_table.py
@@ -379,7 +379,7 @@ def _get_tasks(self, num: int,
                     "sticky": sticky
                 }
             tables = sorted(state_dict)
-            # rng.shuffle(tables)  # type: ignore
+            rng.shuffle(tables)  # type: ignore
             target_table = tables[-1]
             cube_table, cup_table  = tables[:2]
             # Create cube.

diff --git a/scripts/plotting/create_active_sampler_learning_plots.py b/scripts/plotting/create_active_sampler_learning_plots.py
@@ -126,8 +126,8 @@ def _derive_per_task_average(metric: str,
             lambda v: "sticky_table-success_rate_explore" in v)),
         ("Task-Relevant", "purple", lambda df: df["EXPERIMENT_ID"].apply(
             lambda v: "sticky_table-random_score_explore" in v)),
-        ("Random Skills", "blue", lambda df: df["EXPERIMENT_ID"].apply(
-            lambda v: "sticky_table-random_nsrts_explore" in v)),
+        # ("Random Skills", "blue", lambda df: df["EXPERIMENT_ID"].apply(
+        #     lambda v: "sticky_table-random_nsrts_explore" in v)),
         # ("Maple Q", "gray", lambda df: df["EXPERIMENT_ID"].apply(
         #     lambda v: "sticky_table-maple_q" in v)),
     ],