fix: fix ensemble load error

jung235 · Mar 6, 2024 · 702341d · 702341d
1 parent cd76733
commit 702341d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: