Minor fix

PySport · Aug 21, 2024 · 017af5f · 017af5f
1 parent 2b1762a
commit 017af5f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/ingestify/infra/store/dataset/sqlalchemy/mapping.py b/ingestify/infra/store/dataset/sqlalchemy/mapping.py
@@ -23,6 +23,7 @@
 class TZDateTime(TypeDecorator):
     impl = DateTime
     LOCAL_TIMEZONE = datetime.datetime.utcnow().astimezone().tzinfo
+    cache_ok = True
 
     def process_bind_param(self, value: datetime, dialect):
         if value.tzinfo is None:

diff --git a/ingestify/infra/store/dataset/sqlalchemy/repository.py b/ingestify/infra/store/dataset/sqlalchemy/repository.py
@@ -217,7 +217,7 @@ def apply_query_filter(query):
                 func.min(File.modified_at).label("first_modified_at"),
                 func.max(File.modified_at).label("last_modified_at"),
                 func.count().label("row_count"),
-            )
+            ).join(Dataset, Dataset.dataset_id == File.dataset_id)
         ).first()
         dataset_collection_metadata = DatasetCollectionMetadata(*metadata_result_row)
 

diff --git a/ingestify/utils.py b/ingestify/utils.py
@@ -230,4 +230,5 @@ def run(self, func, iterable):
         )
 
     def join(self):
+        self.pool.close()
         self.pool.join()