diff --git a/setup.cfg b/setup.cfg index ebaf825352..0856b359e6 100644 --- a/setup.cfg +++ b/setup.cfg @@ -11,3 +11,9 @@ exclude = omit = test/* setup.py + +[mypy] +exclude = ^.*fb.*$ + +[mypy-captum.log.fb.*] +ignore_errors = True diff --git a/tests/influence/_core/test_tracin_regression.py b/tests/influence/_core/test_tracin_regression.py index 9609091698..90708d4b8d 100644 --- a/tests/influence/_core/test_tracin_regression.py +++ b/tests/influence/_core/test_tracin_regression.py @@ -376,6 +376,7 @@ def _test_tracin_identity_regression_setup( DataInfluenceConstructor( ArnoldiInfluenceFunction, arnoldi_tol=1e-8, # needs to be small to avoid empty arnoldi basis + hessian_reg=2e-3, ), ), ],