diff --git a/sinergym/utils/callbacks.py b/sinergym/utils/callbacks.py index 37b67cde02..0bbbc5bfcc 100644 --- a/sinergym/utils/callbacks.py +++ b/sinergym/utils/callbacks.py @@ -9,8 +9,8 @@ from stable_baselines3.common.env_util import is_wrapped from stable_baselines3.common.vec_env import VecEnv, sync_envs_normalization -from sinergym.utils.wrappers import LoggerWrapper, NormalizeObservation from sinergym.utils.evaluation import evaluate_policy +from sinergym.utils.wrappers import LoggerWrapper, NormalizeObservation class LoggerCallback(BaseCallback): diff --git a/sinergym/utils/evaluation.py b/sinergym/utils/evaluation.py index 28599245b9..ac0a8a0f1c 100644 --- a/sinergym/utils/evaluation.py +++ b/sinergym/utils/evaluation.py @@ -1,7 +1,7 @@ """Custom policy evaluations for Evaluation Callbacks.""" -from itertools import accumulate import warnings +from itertools import accumulate from typing import Any, Callable, Dict, Optional, Union import gym