File tree 1 file changed +3
-8
lines changed
1 file changed +3
-8
lines changed Original file line number Diff line number Diff line change @@ -37,12 +37,7 @@ def __init__(self, series: bigframes.series.Series):
37
37
self ._series = series
38
38
39
39
def __getitem__ (self , key ) -> bigframes .series .Series :
40
- """
41
- Only indexing by a boolean bigframes.series.Series or list of index entries is currently supported
42
- """
43
- return typing .cast (
44
- bigframes .series .Series , _loc_getitem_series_or_dataframe (self ._series , key )
45
- )
40
+ return _loc_getitem_series_or_dataframe (self ._series , key )
46
41
47
42
def __setitem__ (self , key , value ) -> None :
48
43
# TODO(swast): support MultiIndex
@@ -261,8 +256,8 @@ def _loc_getitem_series_or_dataframe(
261
256
return _perform_loc_list_join (series_or_dataframe , keys_df )
262
257
else :
263
258
raise TypeError (
264
- "Invalid argument type. loc currently only supports indexing with a "
265
- "boolean bigframes Series, a list of index entries or a single index entry . "
259
+ "Invalid argument type. Expected bigframes.Series, bigframes.Index, "
260
+ "list, : (empty slice), or scalar . "
266
261
f"{ constants .FEEDBACK_LINK } "
267
262
)
268
263
You can’t perform that action at this time.
0 commit comments