Update utils_ate_bounds.py

jaabmar · Apr 25, 2024 · f43e839 · f43e839
1 parent db1cdf4
commit f43e839
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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