From 702341ddaf54911875b00b37fcfc030d208a1f89 Mon Sep 17 00:00:00 2001
From: jung235 <jung65537@gmail.com>
Date: Wed, 6 Mar 2024 16:01:28 +0900
Subject: [PATCH] fix: fix ensemble load error

---
 pydiffuser/tracer/ensemble.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pydiffuser/tracer/ensemble.py b/pydiffuser/tracer/ensemble.py
index 261d126..2eccea2 100644
--- a/pydiffuser/tracer/ensemble.py
+++ b/pydiffuser/tracer/ensemble.py
@@ -193,7 +193,7 @@ def get_velocity_autocorrelation(
     def __reduce__(self) -> Tuple[Any, ...]:
         info = super().__reduce__()
         args = self.dt
-        new_info = (self.__class__, args, *info[2:])
+        new_info = (self.__class__, (args,), *info[2:])
         return new_info
 
     def to_npy(self, npy_path: PathType) -> None: