add dual updater arg

Saurav Agarwal · Saurav Agarwal · commit 793633e8dae3 · 2025-03-20T14:38:46.000-04:00
diff --git a/python/evaluators/constrained_learning.py b/python/evaluators/constrained_learning.py
@@ -299,6 +299,7 @@ def fun_dual_updater(self, configs, lambdas):
     T_0s = [25]
     eta_duals = [1]
     eval_dir = sys.argv[2]
+    dual_updater = sys.argv[3]
 
     for eta_dual in eta_duals:
         for T_0 in T_0s:
@@ -309,7 +310,7 @@ def fun_dual_updater(self, configs, lambdas):
                     env_id,
                     eta_dual,
                     T_0,
-                    dual_updater="proj_1",
+                    dual_updater,
                     alpha=0.0,
                     normalize=True,
                     obj_normalize_factor=1e10,
