World.restore_path: Add value check

2023-10-18 12:33:30 +02:00 · 2023-10-18 12:33:30 +02:00 · 80b1b1ff9d
commit 80b1b1ff9d
parent c7d0cbb5d3
1 changed files with 4 additions and 0 deletions
--- a/tianshou/highlevel/world.py
+++ b/tianshou/highlevel/world.py
@ -27,4 +27,8 @@ class World:
        return os.path.join(self.persist_directory, filename)

    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)