Remove persist object

NeuralMMO · Oct 13, 2023 · 15b1fb1 · 15b1fb1
1 parent 26f64fe
commit 15b1fb1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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():