Skip to content

Commit cfb043c

Browse files
committed
fix tests
1 parent 9fa8d5c commit cfb043c

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

mapreader/spot_text/runner_base.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -711,8 +711,8 @@ def explore_predictions(
711711

712712
preds_df = self._dict_to_dataframe(self.geo_predictions)
713713
preds_df.drop(
714-
columns=["pixel_geometry"], inplace=True
715-
) # drop pixel_geometry since we can't have two polygon columns
714+
columns=["pixel_geometry", "pixel_line", "line"], inplace=True
715+
) # drop other geometry since we can't have more than one geometry columns
716716

717717
return preds_df[preds_df["image_id"] == parent_id].explore(
718718
tiles=tiles,
@@ -1163,8 +1163,8 @@ def explore_search_results(
11631163
geo_search_results = self._get_geo_search_results()
11641164
geo_df = self._dict_to_dataframe(geo_search_results)
11651165
geo_df.drop(
1166-
columns=["pixel_geometry"], inplace=True
1167-
) # drop pixel_geometry since we can't have two polygon columns
1166+
columns=["pixel_geometry", "pixel_line", "line"], inplace=True
1167+
) # drop other geometry since we can't have more than one geometry columns
11681168

11691169
return geo_df[geo_df["image_id"] == parent_id].explore(
11701170
tiles=tiles,

0 commit comments

Comments
 (0)