diff --git a/tests/gpu_tests/prepare_tests.py b/tests/gpu_tests/prepare_tests.py index 01fc75b..d13753f 100644 --- a/tests/gpu_tests/prepare_tests.py +++ b/tests/gpu_tests/prepare_tests.py @@ -98,6 +98,7 @@ def run_analysis() -> None: activation_covariance_dtype=torch.float64, gradient_covariance_dtype=torch.float64, lambda_dtype=torch.float64, + lambda_iterative_aggregate=True, ) analyzer.fit_all_factors( factors_name="single_gpu",