Updating notebook

Author: JeanVanDyk

causalpy/tests/test_integration_pymc_examples.py

@@ -423,7 +423,7 @@ def test_its_no_treatment_time():
     treatment_time = (pd.to_datetime("2014-01-01"), pd.to_datetime("2022-01-01"))
 
     # Assert that we correctfully raise a value if the given model can't predict InterventionTime
-    with pytest.raises(ValueError) as exc_info:
+    with pytest.raises(cp.custom_exceptions.ModelException) as exc_info:
         cp.InterruptedTimeSeries(
             df,
             treatment_time,