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 "