Skip to content

Commit 8d12c64

Browse files
committed
Format with black
1 parent 2ebbd78 commit 8d12c64

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/reward_preprocessing/trainers/supervised_trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,9 @@ def _record_dataset_stats(self, name: str, dataloader: data.DataLoader) -> None:
309309
self.logger.record(
310310
f"{name}/obs_mean_channel{channel_i}_{name}", obs_mean[channel_i]
311311
)
312-
self.logger.record(f"{name}/obs_std_channel{channel_i}_{name}", obs_std[channel_i])
312+
self.logger.record(
313+
f"{name}/obs_std_channel{channel_i}_{name}", obs_std[channel_i]
314+
)
313315
self.logger.record(f"{name}/rew_mean_{name}", rew_mean)
314316
self.logger.record(f"{name}/rew_std_{name}", rew_std)
315317
self.logger.record(f"{name}/rew_hist_{name}", rew_hist)

0 commit comments

Comments
 (0)