Update image_utils.py

Co-authored-by: NullSenseStudio <[email protected]>
carson-katri · Jun 3, 2024 · 147230a · 147230a
1 parent 8779c3b
commit 147230a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/image_utils.py b/image_utils.py
@@ -689,7 +689,10 @@ def np_to_render_pass(
         array = to_dtype(array, dtype)
     if top_to_bottom:
         array = np.flipud(array)
-    render_pass.rect.foreach_set(array.reshape(-1))
+    if BLENDER_VERSION >= (4, 1, 0):
+        render_pass.rect.foreach_set(array.reshape(-1))
+    else:
+        render_pass.rect.foreach_set(array.reshape(-1, render_pass.channels))
 
 
 def _mode(array, mode):