From 889f172cb27819f771681c91c9b51f8f1e132a17 Mon Sep 17 00:00:00 2001
From: Joseph Suarez <sealsuarez@gmail.com>
Date: Sat, 14 Oct 2023 01:58:23 +0000
Subject: [PATCH] Fix training with new ranker. You may have to clear out your
 old data dir

---
 clean_pufferl.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/clean_pufferl.py b/clean_pufferl.py
index 262f907c..36d9473c 100644
--- 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():