diff --git a/pipestat/exceptions.py b/pipestat/exceptions.py index 3bc00c1e..a13c2d18 100644 --- a/pipestat/exceptions.py +++ b/pipestat/exceptions.py @@ -19,6 +19,7 @@ "InvalidTimeFormatError", "PipestatDependencyError", "ColumnNotFoundError", + "SchemaValidationErrorDuringReport", ] diff --git a/tests/test_pipestat.py b/tests/test_pipestat.py index 1015e418..dd211663 100644 --- a/tests/test_pipestat.py +++ b/tests/test_pipestat.py @@ -7,7 +7,6 @@ import pytest from jsonschema import ValidationError -import pipestat.exceptions from pipestat import SamplePipestatManager, ProjectPipestatManager, PipestatBoss, PipestatManager from pipestat.const import * from pipestat.exceptions import * @@ -385,7 +384,7 @@ def test_complex_object_report_missing_fields( psm = SamplePipestatManager(**args) del val[list(val.keys())[0]]["path"] - with pytest.raises(pipestat.exceptions.SchemaValidationErrorDuringReport): + with pytest.raises(SchemaValidationErrorDuringReport): psm.report(record_identifier=REC_ID, values=val, force_overwrite=True) @pytest.mark.parametrize(