Use max_header_size param only for numpy >= 1.24.0

gem · Oct 11, 2023 · c94f5f7 · c94f5f7
1 parent a47ba11
commit c94f5f7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/svir/tasks/extract_npz_task.py b/svir/tasks/extract_npz_task.py
@@ -178,9 +178,13 @@ def run(self):
                 "%s: returned an empty content" % err_msg))
             return
         try:
-            extracted_npz = numpy.load(
-                io.BytesIO(resp.content), allow_pickle=False,
-                max_header_size=100000)
+            if numpy.__version__ >= '1.24.0':
+                extracted_npz = numpy.load(
+                    io.BytesIO(resp.content), allow_pickle=False,
+                    max_header_size=100000)
+            else:
+                extracted_npz = numpy.load(
+                    io.BytesIO(resp.content), allow_pickle=False)
         except Exception as exc:
             self.exception_sig.emit(exc)
             return