hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

World.restore_path: Add value check

Browse Source
This commit is contained in:
Dominik Jain 2023-10-18 12:33:30 +02:00
parent c7d0cbb5d3
commit 80b1b1ff9d
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tianshou/highlevel/world.py
View File

@ -27,4 +27,8 @@ class World:
return os.path.join(self.persist_directory, filename) return os.path.join(self.persist_directory, filename)
def restore_path(self, filename: str) -> str: def restore_path(self, filename: str) -> str:
if self.restore_directory is None:
raise ValueError(
"Path cannot be formed because no directory for restoration was provided",
)
return os.path.join(self.restore_directory, filename) return os.path.join(self.restore_directory, filename)