fix input preparation

fish-quant · Aug 7, 2020 · 94a7264 · 94a7264
1 parent e36c318
commit 94a7264
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bigfish/classification/input_preparation.py b/bigfish/classification/input_preparation.py
@@ -185,7 +185,7 @@ def prepare_extracted_data(cell_mask, nuc_mask=None, ndim=None, rna_coord=None,
 
         # build distance map from centroid
         if len(centrosome_coord) == 0:
-            distance_centrosome = np.array([0, 0], dtype=np.int64)
+            distance_centrosome = distance_cell.copy()
         else:
             distance_centrosome = _get_centrosome_distance_map(
                 centrosome_coord, cell_mask)