Fix training with new ranker. You may have to clear out your old data…

… dir
PufferAI · Oct 14, 2023 · 889f172 · 889f172
1 parent c1a4357
commit 889f172
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/clean_pufferl.py b/clean_pufferl.py
@@ -144,9 +144,8 @@ def __post_init__(self, *args, **kwargs):
         # Create policy ranker
         if self.policy_ranker is None:
             if self.data_dir is not None:
-                self.policy_ranker = pufferlib.utils.PersistentObject(
+                self.policy_ranker = pufferlib.policy_ranker.OpenSkillRanker(
                     os.path.join(self.data_dir, "openskill.pickle"),
-                    pufferlib.policy_ranker.OpenSkillRanker,
                     "anchor",
                 )
             if "learner" not in self.policy_ranker.ratings():