chore: replace literal feature_id with constant

kraina-ai · Feb 22, 2023 · daae82b · daae82b
1 parent 46a677c
commit daae82b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/srai/loaders/osm_tag_loader/osm_tag_loader.py b/srai/loaders/osm_tag_loader/osm_tag_loader.py
@@ -12,7 +12,7 @@
 from functional import seq
 from tqdm import tqdm
 
-from srai.utils.constants import WGS84_CRS
+from srai.utils.constants import FEATURES_INDEX, WGS84_CRS
 
 
 class OSMTagLoader:
@@ -121,9 +121,9 @@ def _get_empty_result(self) -> gpd.GeoDataFrame:
 
     def _flatten_index(self, gdf: gpd.GeoDataFrame) -> gpd.GeoDataFrame:
         gdf = gdf.reset_index()
-        gdf["feature_id"] = (
+        gdf[FEATURES_INDEX] = (
             gdf[self._RESULT_INDEX_NAMES]
             .apply(lambda idx: "/".join(map(str, idx)), axis=1)
             .astype(str)
         )
-        return gdf.set_index("feature_id").drop(columns=self._RESULT_INDEX_NAMES)
+        return gdf.set_index(FEATURES_INDEX).drop(columns=self._RESULT_INDEX_NAMES)