Skip to content

Commit

Permalink
Automated format fixes (autopep8 + isort)
Browse files Browse the repository at this point in the history
  • Loading branch information
AlejandroCN7 authored and github-actions[bot] committed Sep 19, 2024
1 parent 69ef765 commit 4d0df14
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion sinergym/utils/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -201,6 +201,7 @@ def reset_data(self) -> None:
try:
import wandb
from stable_baselines3.common.logger import KVWriter

class WandBOutputFormat(KVWriter): # pragma: no cover
"""
Dumps key / value pairs onto WandB. This class is based on SB3 used in logger callback
Expand Down Expand Up @@ -247,5 +248,7 @@ def write(
class WandBOutputFormat():
"""WandBOutputFormat class for logging in WandB from SB3 logger.
"""

def __init__(self):
print('WandB or SB3 is not installed. Please install it to use WandBOutputFormat.')
print(
'WandB or SB3 is not installed. Please install it to use WandBOutputFormat.')

0 comments on commit 4d0df14

Please sign in to comment.