From 8dd95e3e5927ad07ea7550da40f1b88a7c837634 Mon Sep 17 00:00:00 2001
From: zhiyang7 <zhiyang7@iflytek.com>
Date: Tue, 4 Jan 2022 18:21:13 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DBUG?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 douzero/env/env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/douzero/env/env.py b/douzero/env/env.py
index 98fd9d2..c774018 100644
--- a/douzero/env/env.py
+++ b/douzero/env/env.py
@@ -195,7 +195,7 @@ class Env:
             }
             obs = None
         else:
-            obs = get_obs(self.infoset, self.use_general, self.use_legacy, self.lite_model)
+            obs = get_obs(self.infoset, self.use_general, self.use_legacy, self.lite_model, self.use_unified)
         return obs, reward, done, {}
 
     def _get_reward(self, pos):