fix testing bug

Former-commit-id: c3a34e4
chenmich · Jun 29, 2020 · c3cd933 · c3cd933
1 parent da68d75
commit c3cd933
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/agents/test_deepcfr.py b/tests/agents/test_deepcfr.py
@@ -12,6 +12,7 @@ def test_init(self):
         sess = tf.InteractiveSession()
         env = rlcard.make('leduc-holdem', config={'allow_step_back':True})
         agent = DeepCFR(session=sess,
+                        scope='deepcfr',
                         env=env,
                         policy_network_layers=(4,4),
                         advantage_network_layers=(4,4),
@@ -37,6 +38,7 @@ def test_train(self):
         sess = tf.InteractiveSession()
         env = rlcard.make('leduc-holdem', {'allow_step_back':True})
         agent = DeepCFR(session=sess,
+                        scope='deepcfr',
                         env=env,
                         policy_network_layers=(128,128),
                         advantage_network_layers=(128,128),