From 3b997fb25eda24d54243b7a598ea83beeb1addbc Mon Sep 17 00:00:00 2001 From: Omkar Dige Date: Fri, 15 Mar 2024 12:14:42 -0400 Subject: [PATCH] Minor fix --- kronfluence/computer/covariance_computer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kronfluence/computer/covariance_computer.py b/kronfluence/computer/covariance_computer.py index bf0199b..8dfaa1d 100644 --- 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 "