From eb5be87f811a53981dfe339efdcc4d0c774ece51 Mon Sep 17 00:00:00 2001 From: Wind <116412564+windlgrass@users.noreply.github.com> Date: Thu, 29 Jan 2026 15:16:34 +0700 Subject: [PATCH] Update dataset_env.py --- environments/dataset_environment/dataset_env.py | 1 + 1 file changed, 1 insertion(+) diff --git a/environments/dataset_environment/dataset_env.py b/environments/dataset_environment/dataset_env.py index 23548c73..ecfc9ec2 100644 --- a/environments/dataset_environment/dataset_env.py +++ b/environments/dataset_environment/dataset_env.py @@ -78,6 +78,7 @@ class DatasetEnv(BaseEnv): self.metric_buffer = {} self.reward_function = self._initialize_reward_function() + self.current_item = None def _initialize_reward_function(self): if hasattr(self.config, "reward_functions") and self.config.reward_functions: