Fix avg

Saurav Agarwal · Saurav Agarwal · commit 9132d752d304 · 2024-09-05T20:13:18.000-04:00
diff --git a/python/scripts/evaluators/constrained_learning.py b/python/scripts/evaluators/constrained_learning.py
@@ -283,7 +283,7 @@ def fun_dual_updater(self, configs, lambdas):
             return lambdas
 
         if configs == "avg":
-            self.lambda_duals = np.array([1.0 / self.num_idfs for i in range(self.num_idfs)])
+            lambdas = np.array([1.0 / self.num_idfs for i in range(self.num_idfs)])
             return lambdas
 
         raise ValueError("configs not recognized")
@@ -308,7 +308,7 @@ def fun_dual_updater(self, configs, lambdas):
                     env_id,
                     eta_dual,
                     T_0,
-                    dual_updater="proj_1",
+                    dual_updater="avg",
                     alpha=0.0,
                     normalize=True,
                     obj_normalize_factor=1e10,
