From e108d232f3941af1250be847b30782718b43da48 Mon Sep 17 00:00:00 2001 From: Mark McDonald Date: Fri, 10 Nov 2023 15:34:20 -0800 Subject: [PATCH] Allow `refresh()` to take args (#95) `refresh()` can be called with args, so accept arbitrary args for the fake implementation. This was an issue internally, where refresh appears to be called with args. --- tests/test_operations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_operations.py b/tests/test_operations.py index 281cc0dac..148b8a2f8 100644 --- a/tests/test_operations.py +++ b/tests/test_operations.py @@ -64,7 +64,7 @@ def test_end_to_end(self): # Create the operation with the `initial_pb` but when it asks for an update # return the `final_pb`. - def refresh(): + def refresh(*_, **__): return final_pb # This is the base Operation class @@ -133,7 +133,7 @@ def make_metadata(completed_steps): ops = gen_operations() initial_pb = next(ops) - def refresh(): + def refresh(*_, **__): """get the next status on each refresh""" return next(ops)