saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-31 23:28:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Ignore pytype error (#1623)

Browse source
This commit is contained in:
Antonin RAFFIN 2023-07-24 14:38:22 +02:00 committed by GitHub
parent 5abd50a853
commit 72c124d907
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
stable_baselines3/common/monitor.py
View file

@ -193,7 +193,9 @@ class ResultsWriter:
mode = "w" if override_existing else "a"
# Prevent newline issue on Windows, see GH issue #692
self.file_handler = open(filename, f"{mode}t", newline="\n")
self.logger = csv.DictWriter(self.file_handler, fieldnames=("r", "l", "t", *extra_keys))
self.logger = csv.DictWriter(
self.file_handler, fieldnames=("r", "l", "t", *extra_keys)
) # pytype: disable=wrong-arg-types
if override_existing:
self.file_handler.write(f"#{json.dumps(header)}\n")
self.logger.writeheader()