From 551100f27b0c6fd57cb345899453e4e7a1c2302a Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Thu, 6 Aug 2026 08:10:21 -0700 Subject: [PATCH] add pyrefly suppressions for jax.jit annotation change PiperOrigin-RevId: 960309017 --- flow_field.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flow_field.py b/flow_field.py index 7669eea..3b1f963 100644 --- a/flow_field.py +++ b/flow_field.py @@ -574,7 +574,7 @@ def flow_field( if pre_mask is not None: s = geom_utils.query_integral_image( - _integral_image(pre_mask), patch_size, step + _integral_image(pre_mask), patch_size, step # pyrefly: ignore[bad-argument-type] ) m = (s / np.prod(patch_size) >= max_masked)[out_sel] selection_mask[m] = False @@ -582,7 +582,7 @@ def flow_field( if post_mask is not None: s = geom_utils.query_integral_image( - _integral_image(post_mask), post_patch_size, step + _integral_image(post_mask), post_patch_size, step # pyrefly: ignore[bad-argument-type] ) m = (s / np.prod(post_patch_size) >= max_masked)[out_sel] selection_mask[m] = False