Merge pull request #822 from MolSSI/cache_fix

Fix missing client on cached dataset records
MolSSI · Apr 24, 2024 · ca892ba · ca892ba
2 parents e1ba960 + ae8fc9e
commit ca892ba
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/qcportal/qcportal/dataset_models.py b/qcportal/qcportal/dataset_models.py
@@ -1045,6 +1045,9 @@ def fetch_records(
 
                     # Check if we have any cached records
                     cached_records = self._cache_data.get_dataset_records(missing_entries, [spec_name])
+                    for _, _, cr in cached_records:
+                        cr.propagate_client(self._client)
+
                     records_batch.extend(cached_records)
 
                     # what we need to fetch from the server
@@ -1171,6 +1174,9 @@ def iterate_records(
 
                         # Check if we have any cached records
                         cached_records = self._cache_data.get_dataset_records(missing_entries, [spec_name])
+                        for _, _, cr in cached_records:
+                            cr.propagate_client(self._client)
+
                         records_batch.extend(cached_records)
 
                         # what we need to fetch from the server