From 15b1fb1c9f6acf9670cdbc2558563318f2db0fe0 Mon Sep 17 00:00:00 2001
From: Joseph Suarez <sealsuarez@gmail.com>
Date: Fri, 13 Oct 2023 03:13:53 +0000
Subject: [PATCH] Remove persist object

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

diff --git a/reinforcement_learning/clean_pufferl.py b/reinforcement_learning/clean_pufferl.py
index 262f907..36d9473 100644
--- a/reinforcement_learning/clean_pufferl.py
+++ b/reinforcement_learning/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():