File tree Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Expand file tree Collapse file tree 2 files changed +3
-8
lines changed Original file line number Diff line number Diff line change @@ -162,14 +162,13 @@ def read_gbq(
162
162
"""
163
163
pandas_gbq = _try_import ()
164
164
165
- kwargs : Dict [str , Union [str , bool ]] = {}
165
+ kwargs : Dict [str , Union [str , bool , None ]] = {}
166
166
167
167
# START: new kwargs. Don't populate unless explicitly set.
168
168
if use_bqstorage_api is not None :
169
169
kwargs ["use_bqstorage_api" ] = use_bqstorage_api
170
170
171
- if progress_bar_type is not None :
172
- kwargs ["progress_bar_type" ] = progress_bar_type
171
+ kwargs ["progress_bar_type" ] = progress_bar_type
173
172
# END: new kwargs
174
173
175
174
return pandas_gbq .read_gbq (
Original file line number Diff line number Diff line change @@ -142,11 +142,7 @@ def mock_read_gbq(sql, **kwargs):
142
142
143
143
monkeypatch .setattr ("pandas_gbq.read_gbq" , mock_read_gbq )
144
144
pd .read_gbq ("SELECT 1" , progress_bar_type = progress_bar )
145
-
146
- if progress_bar :
147
- assert "progress_bar_type" in captured_kwargs
148
- else :
149
- assert "progress_bar_type" not in captured_kwargs
145
+ assert "progress_bar_type" in captured_kwargs
150
146
151
147
152
148
@pytest .mark .single
You can’t perform that action at this time.
0 commit comments