diff --git a/python/pyarrow/tests/test_flight.py b/python/pyarrow/tests/test_flight.py index 2c53e4f63f2..db26f0df4fa 100644 --- a/python/pyarrow/tests/test_flight.py +++ b/python/pyarrow/tests/test_flight.py @@ -57,6 +57,8 @@ def __exit__(self, exc_type, exc_value, traceback): flight = None class MockContextManager: + def __init__(self, *args, **kwargs): + pass def __enter__(self): return self def __exit__(self, exc_type, exc_val, exc_tb):