Added Sheriff tests(Look into initial iterations)

Jamesflynn1 · Nov 15, 2023 · c4cd092 · c4cd092
1 parent e5b8ed4
commit c4cd092
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/open_spiel/python/algorithms/efr_test.py b/open_spiel/python/algorithms/efr_test.py
@@ -90,16 +90,18 @@ def test_efr_kuhn_poker_3p(self, deviations_name):
 
 
   @parameterized.parameters(
-      ["blind cf", "informed cf", "bps", "cfps", "csps"])
+      ["blind cf", "bps", "tips"])
   def test_efr_cce_dist_sheriff(self, deviations_name):
     game = pyspiel.load_game("sheriff")
     efr_solver = efr.EFRSolver(
         game=game,
         deviations_name=deviations_name
-    )    
+    )   
+    #efr_solver = cfr.CFRSolver(game)
+
     strategies = []
     corr_dist_values = []
-    for _ in range(3):
+    for _ in range(5):
       efr_solver.evaluate_and_update_policy()
       strategies.append(policy.python_policy_to_pyspiel_policy(
           efr_solver.current_policy()))