From 0fdebcbb760ac94b75bb755ab89647fe00c245da Mon Sep 17 00:00:00 2001 From: phonixer <35895359+phonixer@users.noreply.github.com> Date: Thu, 18 Jul 2024 13:31:37 +0800 Subject: [PATCH] Update buffer.py Correct np.int -> int --- diffuser/datasets/buffer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffuser/datasets/buffer.py b/diffuser/datasets/buffer.py index 1ad2106b..71733298 100644 --- a/diffuser/datasets/buffer.py +++ b/diffuser/datasets/buffer.py @@ -9,7 +9,7 @@ class ReplayBuffer: def __init__(self, max_n_episodes, max_path_length, termination_penalty): self._dict = { - 'path_lengths': np.zeros(max_n_episodes, dtype=np.int), + 'path_lengths': np.zeros(max_n_episodes, dtype=int), } self._count = 0 self.max_n_episodes = max_n_episodes