Avoid probabilities calculation if parameter False

marcos-diazg · marcos-diazg · commit cf1ccae9c8f8 · 2022-10-17T13:54:00.000-07:00
diff --git a/SigProfilerAssignment/decompose_subroutines.py b/SigProfilerAssignment/decompose_subroutines.py
@@ -772,10 +772,11 @@ def make_final_solution(processAvg, allgenomes, allsigids, layer_directory, m, i
         else:
             custom_signatures_plot(processes, layer_directory+"/Signatures")
     
-    probability = probabilities(processAvg, exposureAvg, index, allsigids, allcolnames)
-    probability=probability.set_index("Sample Names" )
-    
     if export_probabilities==True:
+
+        probability = probabilities(processAvg, exposureAvg, index, allsigids, allcolnames)
+        probability=probability.set_index("Sample Names" )
+    
         if cosmic_sigs==False:
             
             if refit_denovo_signatures==True:
