diff --git a/aix360/algorithms/ecertify/ecertify.py b/aix360/algorithms/ecertify/ecertify.py index 4b56027..b15d9e8 100644 --- a/aix360/algorithms/ecertify/ecertify.py +++ b/aix360/algorithms/ecertify/ecertify.py @@ -6,7 +6,7 @@ from algorithms.ecertify.ExpCertifyBB import Ecertify -class CertifyExplanation(): +class CertifyExplanation(LocalBBExplainer): """ Explanation certification class for black-box probabilistic certification of local explanations. Provides access to :class:`aix360.algorithms.ecertify.ExpCertifyBB`, where the certification strategies diff --git a/tests/ecertify/test_ecertify.py b/tests/ecertify/test_ecertify.py index 4a5e88a..cd3f7b9 100644 --- a/tests/ecertify/test_ecertify.py +++ b/tests/ecertify/test_ecertify.py @@ -2,6 +2,7 @@ import numpy as np from aix360.algorithms.ecertify.ExpCertifyBB import Ecertify +from aix360.algorithms.ecertify.ecertify import CertifyExplanation class TestCertifyExplainer(unittest.TestCase):