Minor fix

pomonam · Mar 15, 2024 · 3b997fb · 3b997fb
1 parent 78c2803
commit 3b997fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kronfluence/computer/covariance_computer.py b/kronfluence/computer/covariance_computer.py
@@ -298,10 +298,10 @@ def aggregate_covariance_matrices(
         data_partition_size = factor_args.covariance_data_partition_size
         module_partition_size = factor_args.covariance_module_partition_size
         all_required_partitions = [(i, j) for i in range(data_partition_size) for j in range(module_partition_size)]
-        all_partition_exists = [
+        all_partition_exists = all([
             covariance_matrices_exist(output_dir=factors_output_dir, partition=partition)
             for partition in all_required_partitions
-        ]
+        ])
         if not all_partition_exists:
             self.logger.info(
                 "Covariance matrices are not aggregated as covariance matrices for some partitions "