NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update dataset_env.py

Browse source
This commit is contained in:
Wind 2026-01-29 15:16:34 +07:00 • committed by GitHub
parent 6c2f1ac408
commit eb5be87f81
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
environments/dataset_environment/dataset_env.py
View file

@ -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: