Merge pull request #136 from apike02/ewmv_modified

Ewmv modified
CCS-Lab · Jul 29, 2021 · cf9a08f · cf9a08f
2 parents 8cfc542 + 7036b85
commit cf9a08f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/commons/stan_files/bart_ewmv.stan b/commons/stan_files/bart_ewmv.stan
@@ -94,7 +94,9 @@ model {
       n_succ += pumps[j, k] - explosion[j, k];
       n_pump += pumps[j, k];
 
-      p_burst = phi[j] + (1 - exp(-n_pump * eta[j])) * ((0.0 + n_pump - n_succ) / n_pump - phi[j]);
+      if(n_pump>0){
+        p_burst = phi[j] + (1 - exp(-n_pump * eta[j])) * ((0.0 + n_pump - n_succ) / n_pump - phi[j]);
+      }
     }
   }
 }
@@ -153,7 +155,9 @@ generated quantities {
         n_succ += pumps[j, k] - explosion[j, k];
         n_pump += pumps[j, k];
 
-        p_burst = phi[j] + (1 - exp(-n_pump * eta[j])) * ((0.0 + n_pump - n_succ) / n_pump - phi[j]);
+        if(n_pump>0){
+          p_burst = phi[j] + (1 - exp(-n_pump * eta[j])) * ((0.0 + n_pump - n_succ) / n_pump - phi[j]);
+        }
       }
     }
   }