Adding type suppressions for pyrefly

PiperOrigin-RevId: 950703735
Change-Id: I6b6d5826a0dd1f4c0a1cbb2a46aee944cd60d884
This commit is contained in:
Hana Joo
2026-07-20 02:39:22 -07:00
committed by Copybara-Service
parent 00abc5465f
commit 78946ca94e
30 changed files with 193 additions and 193 deletions
+3 -3
View File
@@ -180,7 +180,7 @@ class Rollout:
model = [model] # Use a length 1 list to simplify code below
if not isinstance(data, list):
data = [data] # Use a length 1 list to simplify code below
data = [data] # Use a length 1 list to simplify code below # pyrefly: ignore[bad-assignment]
# infer nstep, check for incompatibilities
nstep = _infer_dimension(
@@ -310,9 +310,9 @@ def rollout(
ValueError: bad shapes or sizes.
""" # fmt: skip
if not isinstance(data, list):
data = [data] # Use a length 1 list to simplify code below
data = [data] # Use a length 1 list to simplify code below # pyrefly: ignore[bad-assignment]
nthread = len(data) if len(data) > 1 else 0
nthread = len(data) if len(data) > 1 else 0 # pyrefly: ignore[bad-argument-type]
# Use a persistent thread pool if requested
if persistent_pool: