diff --git a/cpa/profiling/profile_mean.py b/cpa/profiling/profile_mean.py index 5eabf255..ef2781a6 100755 --- a/cpa/profiling/profile_mean.py +++ b/cpa/profiling/profile_mean.py @@ -24,7 +24,7 @@ def _compute_group_mean((cache_dir, images, normalization_name, normalization = normalizations[normalization_name] data, colnames, _ = cache.load(images, normalization=normalization) - cellcount = data.shape[0] + cellcount = np.ones(1) * data.shape[0] if method == 'cellcount': return cellcount @@ -109,7 +109,7 @@ def profile_mean(cache_dir, group_name, filter=None, parallel=Uniprocessing(), elif method == 'mean+deciles': variables = variables + ['decile_%02d_%s' % (dec, v) for dec in range(10,100,10) for v in variables] elif method == 'cellcount': - variables = 'Cells_Count' + variables = ['Cells_Count'] return Profiles.compute(keys, variables, _compute_group_mean, parameters, parallel=parallel, group_name=group_name, show_progress=show_progress,