Suppress type errors for Pyrefly.

cascades/_src/distributions/gpt.py

@@ -33,7 +33,7 @@
 @functools.lru_cache()
 def cached_completion(rng=None, **kwargs):
   del rng
-  return openai.Completion.create(**kwargs)
+  return openai.Completion.create(**kwargs)  # pyrefly: ignore[missing-attribute]
 
 
 @dataclasses.dataclass(eq=True, frozen=True)

cascades/examples/calculator.py

@@ -52,9 +52,9 @@ def _eval_arithmetic(node):
   if isinstance(node, ast.Num):  # <number>
     return node.n
   elif isinstance(node, ast.BinOp):  # <left> <operator> <right>
-    return operators[type(node.op)](_eval_arithmetic(node.left),
-                                    _eval_arithmetic(node.right))
+    return operators[type(node.op)](_eval_arithmetic(node.left),  # pyrefly: ignore[bad-index]
+                                    _eval_arithmetic(node.right))  # pyrefly: ignore[bad-argument-count]
   elif isinstance(node, ast.UnaryOp):  # <operator> <operand> e.g., -1
-    return operators[type(node.op)](_eval_arithmetic(node.operand))
+    return operators[type(node.op)](_eval_arithmetic(node.operand))  # pyrefly: ignore[bad-argument-count, bad-index, no-matching-overload]
   else:
     raise TypeError(node)

cascades/examples/tasks/svamp.py

@@ -137,7 +137,7 @@ def solve_svamp_task(task, train_set, lm=None, check_correct=False):
 
   if check_correct:
     # Did we get the right answer?
-    correct = _score_prediction(model_answer=calculator_output,
+    correct = _score_prediction(model_answer=calculator_output,  # pyrefly: ignore[bad-argument-type]
                                 true_answer=task['Answer'])
     yield cc.log(correct, 'is_correct')
     if not correct: