From f43e839a55d5be8ceece9889bd6aa2371fe6503b Mon Sep 17 00:00:00 2001
From: Javier Abad <79517685+jaabmar@users.noreply.github.com>
Date: Thu, 25 Apr 2024 22:05:46 +0200
Subject: [PATCH] Update utils_ate_bounds.py

---
 src/test_confounding/ate_bounds/utils_ate_bounds.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test_confounding/ate_bounds/utils_ate_bounds.py b/src/test_confounding/ate_bounds/utils_ate_bounds.py
index 39888ed..6dc5dc5 100644
--- a/src/test_confounding/ate_bounds/utils_ate_bounds.py
+++ b/src/test_confounding/ate_bounds/utils_ate_bounds.py
@@ -59,7 +59,7 @@ def get_quantile_regressor(
     # Solve linear quantile regression approximately (no linear program) --> much faster and scales to 100k+ datapoints
     if fast_solver:
         # quant_reg = QuantileLinearRegression(quantile=tau, max_iter=500)
-        quant_reg = QuantileRegressor(quantile=tau, max_iter=500)
+        quant_reg = QuantileRegressor(quantile=tau)
         quant_reg.fit(X, y)
         return quant_reg