From f29ed0efdbefc3283c2ab54776bcb00175ca2d39 Mon Sep 17 00:00:00 2001
From: Jet <taijunjet@hotmail.com>
Date: Fri, 9 Aug 2024 11:42:13 +0900
Subject: [PATCH] additional check

---
 wingman/replay_buffer/utils/listed_dict_to_dicted_list.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wingman/replay_buffer/utils/listed_dict_to_dicted_list.py b/wingman/replay_buffer/utils/listed_dict_to_dicted_list.py
index 4e7a24b..0da4556 100644
--- a/wingman/replay_buffer/utils/listed_dict_to_dicted_list.py
+++ b/wingman/replay_buffer/utils/listed_dict_to_dicted_list.py
@@ -65,7 +65,7 @@ def listed_dict_to_dicted_list(
                 ptr[key_list[-1]] = np.stack(dicted_list, axis=0)
             else:
                 ptr[key_list[-1]] = np.concatenate(dicted_list, axis=0)
-        elif isinstance(dicted_list[0], torch.Tensor) and len(dicted_list):
+        elif isinstance(dicted_list[0], torch.Tensor) and len(dicted_list[0].shape) > 0:
             if stack:
                 ptr[key_list[-1]] = torch.stack(dicted_list, dim=0)
             else: