File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change 2424
2525import 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.
2831try :
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 )
You can’t perform that action at this time.
0 commit comments