From ef8ff9ba835f07e89b21147eef1a5bbcee95262f Mon Sep 17 00:00:00 2001
From: Pierre Schumacher <pierre.schumacher@tuebingen.mpg.de>
Date: Wed, 14 Aug 2024 15:06:00 +0200
Subject: [PATCH] minor update for play and load function

---
 deprl/play.py             | 1 +
 deprl/utils/load_utils.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/deprl/play.py b/deprl/play.py
index 08dfef4..d9fcfa5 100644
--- a/deprl/play.py
+++ b/deprl/play.py
@@ -311,6 +311,7 @@ def play(
     config, checkpoint_path, _ = load_checkpoint(checkpoint_path, checkpoint)
 
     # Get important info from config
+    assert config is not None
     header = header or config["tonic"]["header"]
     agent = agent or config["tonic"]["agent"]
     environment = environment or config["tonic"]["test_environment"]
diff --git a/deprl/utils/load_utils.py b/deprl/utils/load_utils.py
index e93811d..06d36a6 100644
--- a/deprl/utils/load_utils.py
+++ b/deprl/utils/load_utils.py
@@ -44,6 +44,7 @@ def load_checkpoint(checkpoint_path, checkpoint="last"):
     Checkpoint loading for main() function.
     """
     if not os.path.isdir(checkpoint_path):
+        logger.log(f"Checkpoint path is not valid: {checkpoint_path}")
         return None, None, None
 
     path = (