diff --git a/sdks/python/apache_beam/runners/interactive/display/pcoll_visualization.py b/sdks/python/apache_beam/runners/interactive/display/pcoll_visualization.py index bfdfe9b453f..b9735409236 100644 --- a/sdks/python/apache_beam/runners/interactive/display/pcoll_visualization.py +++ b/sdks/python/apache_beam/runners/interactive/display/pcoll_visualization.py @@ -423,8 +423,7 @@ def _display_dataframe(self, data, update=None): with warnings.catch_warnings(): warnings.filterwarnings( "ignore", message=".*DataFrame.applymap has been deprecated.*") - rows = data.applymap(lambda x: str(x)).to_dict('split')['data'] - + rows = data.applymap(lambda x: str(x)).to_dict('split')['data'] # Convert each row into dict where keys are column index in the datatable # to be rendered and values are data from the dataframe. Column index 0 is # left out to hold the int index (not part of the data) from dataframe.