saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-16 22:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

using other_file instead of other_files

Browse source
This commit is contained in:
Noah Dormann 2019-12-05 08:52:26 +01:00
parent 7c8d375bcb
commit 6560ae9952
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
torchy_baselines/common/base_class.py
View file

@ -330,11 +330,11 @@ class BaseRLModel(object):
file_content.seek(0)
params = th.load(file_content)
# check for all other .pth files
other_files = [file_name for file_name in namelist if
other_file = [file_name for file_name in namelist if
os.path.splitext(file_name)[1] == ".pth" and file_name != "params.pth"]
if len(other_files) > 0:
if len(other_file) > 0:
opt_params = dict()
for file in other_files:
for file in other_file:
with file_.open(file, mode="r") as opt_param_file:
# File has to be seekable so load in BytesIO first
file_content = io.BytesIO()