PY39: Define strict zip if not builtin

nipreps · Jul 29, 2024 · b297129 · b297129
1 parent dac204c
commit b297129
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/niworkflows/interfaces/bids.py b/niworkflows/interfaces/bids.py
@@ -28,6 +28,7 @@
 import shutil
 import os
 import re
+import sys
 
 import nibabel as nb
 import numpy as np
@@ -64,6 +65,15 @@
 LOGGER = logging.getLogger("nipype.interface")
 
 
+if sys.version_info < (3, 10):  # PY39
+    builtin_zip = zip
+
+    def zip(*args, strict=False):
+        if strict and any(len(args[0]) != len(arg) for arg in args):
+            raise ValueError("strict_zip() requires all arguments to have the same length")
+        return builtin_zip(*args)
+
+
 def _none():
     return None