Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
[(0.1, LinearRegressionEstimator, [0,1], [0,1], [0,], [1, 2], [False, True], [False,]),])
def test_average_treatment_effect(self, error_tolerance, Estimator,
num_common_causes, num_instruments, num_effect_modifiers,
num_treatments, treatment_is_binary, outcome_is_binary
):
estimator_tester = TestEstimator(error_tolerance, Estimator)
estimator_tester.average_treatment_effect_testsuite(
num_common_causes=num_common_causes,
num_instruments = num_instruments,
num_effect_modifiers = num_effect_modifiers,
num_treatments=num_treatments,
treatment_is_binary=treatment_is_binary,
outcome_is_binary=outcome_is_binary
)