From 766f6fedf27835831f9fb4c5a16a32bb6f6e63f6 Mon Sep 17 00:00:00 2001
From: Molasses <2018063+sleeplessai@users.noreply.github.com>
Date: Mon, 15 Apr 2024 17:32:35 +0800
Subject: [PATCH] Fix imports in Readme

---
 README.md | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 7c83615..025bcd3 100644
--- a/README.md
+++ b/README.md
@@ -235,8 +235,9 @@ from tianshou.highlevel.env import (
 from tianshou.highlevel.experiment import DQNExperimentBuilder, ExperimentConfig
 from tianshou.highlevel.params.policy_params import DQNParams
 from tianshou.highlevel.trainer import (
-    TrainerEpochCallbackTestDQNSetEps,
-    TrainerEpochCallbackTrainDQNSetEps,
+    EpochTestCallbackDQNSetEps,
+    EpochTrainCallbackDQNSetEps,
+    EpochStopCallbackRewardThreshold
 )
 ```