fixed bug and compat

KohlerHECTOR · Jul 15, 2024 · 7478768 · 7478768
1 parent 22bfdb1
commit 7478768
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/interpreter/interpreter.py b/interpreter/interpreter.py
@@ -53,7 +53,7 @@ def __init__(self, oracle, tree_policy, env, data_per_iter=5000, **kwargs):
             isinstance(tree_policy, DTPolicy)
             or isinstance(tree_policy, ObliqueDTPolicy)
         )
-        super().__init__(env=env, **kwargs)
+        AgentWithSimplePolicy.__init__(self, env, **kwargs)
         if not isinstance(self.eval_env, Monitor):
             self.eval_env = Monitor(self.eval_env)
         self._oracle = oracle