enable/disable calibrate "bo bookings" demand

nanqiangyipo · Mar 30, 2017 · ce6e4d5 · ce6e4d5
1 parent f7e752e
commit ce6e4d5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/revpy/mfrm.py b/revpy/mfrm.py
@@ -58,7 +58,8 @@ def estimate_host_level(observed, availability, probs, nofly_prob):
     return demand, spill, recapture
 
 
-def estimate_class_level(observed, availability, probs, nofly_prob):
+def estimate_class_level(observed, availability, probs, nofly_prob,
+                         calibrate=True):
     """ Estimate demand, spill and recapture using multi-flight recapture
     method (MFRM) on class-level
 
@@ -109,8 +110,10 @@ def estimate_class_level(observed, availability, probs, nofly_prob):
                 'recapture': 0
             }
 
-    return calibrate_no_booking(estimates, observed, availability, probs,
-                                host_spill)
+    if calibrate:
+        estimates = calibrate_no_booking(estimates, observed, availability,
+                                         probs, host_spill)
+    return estimates
 
 
 def calibrate_no_booking(estimates, observed, availability, probs, host_spill):