@@ -159,9 +159,8 @@ def __getitem__(self, key):
159
159
)
160
160
161
161
columns = key [1 ]
162
- if isinstance (columns , pd .Series ) and columns .dtype == "bool" :
163
- # TODO(b/340892590): fix type error
164
- columns = df .columns [columns ] # type: ignore
162
+ if isinstance (columns , pd .Series ) and columns .dtype == bool :
163
+ columns = df .columns [typing .cast (pd .Series , columns )]
165
164
166
165
return df [columns ]
167
166
@@ -252,7 +251,7 @@ def __getitem__(self, key: tuple) -> bigframes.core.scalar.Scalar:
252
251
raise ValueError (error_message )
253
252
if len (key ) != 2 :
254
253
raise TypeError (error_message )
255
- block : bigframes .core .blocks .Block = self ._dataframe ._block # type: ignore
254
+ block : bigframes .core .blocks .Block = self ._dataframe ._block
256
255
column_block = block .select_columns ([block .value_columns [key [1 ]]])
257
256
column = bigframes .series .Series (column_block )
258
257
return column .iloc [key [0 ]]
@@ -376,14 +375,14 @@ def _perform_loc_list_join(
376
375
)
377
376
result = result .rename (original_name )
378
377
else :
379
- result = series_or_dataframe ._perform_join_by_index (keys_index , how = "right" ) # type: ignore
378
+ result = series_or_dataframe ._perform_join_by_index (keys_index , how = "right" )
380
379
381
380
if drop_levels and series_or_dataframe .index .nlevels > keys_index .nlevels :
382
381
# drop common levels
383
382
levels_to_drop = [
384
383
name for name in series_or_dataframe .index .names if name in keys_index .names
385
384
]
386
- result = result .droplevel (levels_to_drop ) # type: ignore
385
+ result = result .droplevel (levels_to_drop )
387
386
return result
388
387
389
388
0 commit comments