do not use pickle to ensure interop between different python versions;

koi-learning · Jun 26, 2023 · 6e8fc11 · 6e8fc11
1 parent c6f3898
commit 6e8fc11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/koi_core/data/simple_accessor.py b/koi_core/data/simple_accessor.py
@@ -23,12 +23,12 @@
 @cache
 def get_np(self: Union[SampleDatum, SampleLabel], meta) -> np.ndarray:
     f = BytesIO(self.raw)
-    return np.load(f), meta
+    return np.load(f, allow_pickle=False), meta
 
 
 def set_np(self: Union[SampleDatum, SampleLabel], value: np.ndarray) -> None:
     f = BytesIO()
-    np.save(f, value)
+    np.save(f, value, allow_pickle=False)
     self.raw = f.getvalue()
     setCache(self, 'np', None)