From 66d1cfe1001135961731e7ef1214f47cebe4e15b Mon Sep 17 00:00:00 2001
From: Maciej Majek <46171033+maciejmajek@users.noreply.github.com>
Date: Fri, 30 Aug 2024 14:37:29 +0200
Subject: [PATCH] fix: allow numpy to extend filesize for larger memmaps.

---
 openwakeword/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/openwakeword/data.py b/openwakeword/data.py
index c43da5d..5a7eb7b 100755
--- a/openwakeword/data.py
+++ b/openwakeword/data.py
@@ -785,7 +785,7 @@ def __init__(self,
         self.label_transform_funcs = label_transform_funcs
 
         # Get array mmaps and store their shapes (but load files < 1 GB total size into memory)
-        self.data = {label: np.load(fl, mmap_mode='r') for label, fl in data_files.items()}
+        self.data = {label: np.load(fl, mmap_mode='r+') for label, fl in data_files.items()}
         self.labels = {label: np.load(fl) for label, fl in label_files.items()}
         self.data_counter = {label: 0 for label in data_files.keys()}
         self.original_shapes = {label: self.data[label].shape for label in self.data.keys()}