fix: change wrong parameter in ways grouping

kraina-ai · Jan 2, 2024 · 3d96515 · 3d96515
1 parent 00d4ae7
commit 3d96515
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/quackosm/pbf_file_reader.py b/quackosm/pbf_file_reader.py
@@ -945,7 +945,7 @@ def _get_required_ways_with_linestrings(
 
         with TaskProgressSpinner("Grouping required ways", "26"):
             groups = self._group_ways_ids(
-                ways_ids=osm_parquet_files.ways_filtered_ids,
+                ways_ids=osm_parquet_files.ways_required_ids,
                 destination_dir_path=destination_dir_path,
                 grouped_ways_ids_path=grouped_ways_ids_path,
             )
@@ -1015,7 +1015,7 @@ def _construct_ways_linestrings(
                 SELECT * FROM read_parquet('{current_ways_ids_group_path}/**')
             """)
 
-            current_ways_group_relation = self.connection.sql(f"""
+            current_ways_ids_group_with_nodes_relation = self.connection.sql(f"""
                 SELECT
                     w.id, n.point, w.ref_idx
                 FROM ({osm_parquet_files.ways_with_unnested_nodes_refs.sql_query()}) w
@@ -1025,7 +1025,7 @@ def _construct_ways_linestrings(
                 ON n.id = w.ref
             """)
             current_ways_group_relation = self._save_parquet_file(
-                relation=current_ways_group_relation,
+                relation=current_ways_ids_group_with_nodes_relation,
                 file_path=grouped_ways_path / f"group={group}",
             )