From 4eeb23e2b3c5b76c5b129c97b8dd9eeadeb20770 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Wed, 22 Jul 2026 17:37:41 +0800
Subject: [PATCH] fix: use copy-on-write mmap mode to silence non-writable
 tensor warning

---
 astrai/serialization/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/astrai/serialization/dataset.py b/astrai/serialization/dataset.py
index 84629c0..790f5fa 100644
--- a/astrai/serialization/dataset.py
+++ b/astrai/serialization/dataset.py
@@ -100,7 +100,7 @@ def load_bin(file_path: str) -> Dict[str, List[Tensor]]:
         arr = np.memmap(
             os.path.join(file_path, f"{key}.bin"),
             dtype=info["dtype"],
-            mode="r",
+            mode="c",
             shape=tuple(info["shape"]),
         )
         segments[key] = [torch.from_numpy(arr)]