Skip to content

Commit

Permalink
[BugFix] Remove logger warning
Browse files Browse the repository at this point in the history
  • Loading branch information
matteobettini committed Dec 5, 2023
1 parent 6cb0a50 commit db6696d
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion benchmarl/experiment/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
#

import json
import os
from pathlib import Path
from typing import Dict, List, Optional

Expand Down Expand Up @@ -214,9 +215,12 @@ def log_evaluation(
evaluation_step=total_frames
// self.experiment_config.evaluation_interval,
)
json_file = str(self.json_writer.path)
for logger in self.loggers:
if isinstance(logger, WandbLogger):
logger.experiment.save(str(self.json_writer.path))
logger.experiment.save(
json_file, base_path=os.path.dirname(json_file)
)

self.log(to_log, step=step)
if video_frames is not None:
Expand Down

0 comments on commit db6696d

Please sign in to comment.