Add disqualification check to billing model predict()

Signed-off-by: Jason Chulock <[email protected]>

eemeter/eemeter/models/billing/model.py

@@ -22,7 +22,10 @@
 import numpy as np
 import pandas as pd
 
-from eemeter.eemeter.common.exceptions import DataSufficiencyError
+from eemeter.eemeter.common.exceptions import (
+    DataSufficiencyError,
+    DisqualifiedModelError,
+)
 from eemeter.eemeter.common.warnings import EEMeterWarning
 from eemeter.eemeter.models.billing.data import (
     BillingBaselineData,
@@ -66,9 +69,15 @@ def predict(
         self,
         reporting_data: Union[BillingBaselineData, BillingReportingData],
         aggregation=None,
+        ignore_disqualification=False,
     ):
         if not self.is_fitted:
             raise RuntimeError("Model must be fit before predictions can be made.")
+
+        if self.disqualification and not ignore_disqualification:
+            raise DisqualifiedModelError(
+                "Attempting to predict using disqualified model without setting ignore_disqualification=True"
+            )
 
         if not isinstance(reporting_data, (BillingBaselineData, BillingReportingData)):
             raise TypeError(

tests/test_derivatives.py

@@ -248,7 +248,8 @@ def test_metered_savings_cdd_hdd_billing_single_record_baseline_data(
             reporting_meter_data_billing,
             reporting_temperature_data,
             is_electricity_data=True,
-        )
+        ),
+        ignore_disqualification=True,
     )
     assert list(results.columns) == [
         "season",
@@ -524,7 +525,8 @@ def test_metered_savings_model_single_record(
             reporting_meter_data_billing,
             reporting_temperature_data,
             is_electricity_data=True,
-        )
+        ),
+        ignore_disqualification=True,
     )
     assert list(results.columns) == [
         "season",