Skip to content

Commit df69bb1

Browse files
Henry J Solbergmilkshakeiii
Henry J Solberg
authored andcommitted
bug: change return type of Series.loc[scalar]
Change-Id: Id60a7da3021972da5c8a28fb8f3620e10643c0ed
1 parent 109ee24 commit df69bb1

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

bigframes/core/indexers.py

+3-8
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,7 @@ def __init__(self, series: bigframes.series.Series):
3737
self._series = series
3838

3939
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)
4641

4742
def __setitem__(self, key, value) -> None:
4843
# TODO(swast): support MultiIndex
@@ -261,8 +256,8 @@ def _loc_getitem_series_or_dataframe(
261256
return _perform_loc_list_join(series_or_dataframe, keys_df)
262257
else:
263258
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. "
266261
f"{constants.FEEDBACK_LINK}"
267262
)
268263

0 commit comments

Comments
 (0)