fix custom_info

2022-12-05 20:22:54 +01:00
parent d8565261e1
commit 62c69bf2b5
4 changed files with 4 additions and 6 deletions
--- a/freqtrade/freqai/RL/TensorboardCallback.py
+++ b/freqtrade/freqai/RL/TensorboardCallback.py
@@ -42,7 +42,7 @@ class TensorboardCallback(BaseCallback):
        )

    def _on_step(self) -> bool:
-        custom_info = self.training_env.custom_info
+        custom_info = self.training_env.get_attr("custom_info")[0]
        self.logger.record("_state/position", self.locals["infos"][0]["position"])
        self.logger.record("_state/trade_duration", self.locals["infos"][0]["trade_duration"])
        self.logger.record("_state/current_profit_pct", self.locals["infos"]