diff --git a/cascades/_src/distributions/gpt.py b/cascades/_src/distributions/gpt.py index b2f9de0..6295118 100644 --- a/cascades/_src/distributions/gpt.py +++ b/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) diff --git a/cascades/examples/calculator.py b/cascades/examples/calculator.py index ec6d091..59bb6bf 100644 --- a/cascades/examples/calculator.py +++ b/cascades/examples/calculator.py @@ -52,9 +52,9 @@ def _eval_arithmetic(node): if isinstance(node, ast.Num): # return node.n elif isinstance(node, ast.BinOp): # - 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): # 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) diff --git a/cascades/examples/tasks/svamp.py b/cascades/examples/tasks/svamp.py index f19ec5d..088c90b 100644 --- a/cascades/examples/tasks/svamp.py +++ b/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: