get value from tuple

digital-land · Jan 15, 2025 · 25f7618 · 25f7618
1 parent 753b40a
commit 25f7618
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/digital_land/package/dataset_parquet.py b/digital_land/package/dataset_parquet.py
@@ -391,9 +391,9 @@ def load_entities(self, transformed_parquet_dir, resource_path, organisation_pat
 
         # retrieve entity counnts including and minimum
         min_sql = f"select MIN(entity) FROM parquet_scan('{transformed_parquet_dir}/*.parquet');"
-        min_entity = self.conn.execute(min_sql).fetchone()
+        min_entity = self.conn.execute(min_sql).fetchone()[0]
         max_sql = f"select MAX(entity) FROM parquet_scan('{transformed_parquet_dir}/*.parquet');"
-        max_entity = self.conn.execute(max_sql).fetchone()
+        max_entity = self.conn.execute(max_sql).fetchone()[0]
         total_entities = max_entity - min_entity
         entity_limit = 1000000
         if total_entities > entity_limit: