Skip to content

Commit 59767a0

Browse files
committed
change function name
1 parent dd2c27e commit 59767a0

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

bigframes/display/anywidget.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,10 @@
2424

2525
import bigframes
2626

27-
# Simplified import structure as suggested in review
27+
# anywidget and traitlets are optional dependencies. We don't want the import of this
28+
# module to fail if they aren't installed, though. Instead, we try to limit the surface that
29+
# these packages could affect. This makes unit testing easier and ensures we don't
30+
# accidentally make these required packages.
2831
try:
2932
import anywidget
3033
import traitlets
@@ -123,15 +126,15 @@ def _get_next_batch(self) -> bool:
123126
return False
124127

125128
try:
126-
iterator = self._get_batch_iterator()
129+
iterator = self._batch_iterator()
127130
batch = next(iterator)
128131
self._cached_batches.append(batch)
129132
return True
130133
except StopIteration:
131134
self._all_data_loaded = True
132135
return False
133136

134-
def _get_batch_iterator(self) -> Iterator[pd.DataFrame]:
137+
def _batch_iterator(self) -> Iterator[pd.DataFrame]:
135138
"""Lazily initializes and returns the batch iterator."""
136139
if self._batch_iter is None:
137140
self._batch_iter = iter(self._batches)

0 commit comments

Comments
 (0)