From 84c82426cf49a34fdeffa1364c5e5113334c8ac3 Mon Sep 17 00:00:00 2001
From: komo <66017773+komo135@users.noreply.github.com>
Date: Tue, 8 Mar 2022 14:00:54 +0900
Subject: [PATCH] Update dqn.py

---
 traderl/agent/dqn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/traderl/agent/dqn.py b/traderl/agent/dqn.py
index 41f42d5..4b19d91 100644
--- a/traderl/agent/dqn.py
+++ b/traderl/agent/dqn.py
@@ -100,7 +100,7 @@ def env(self):
         return x, y, atr
 
     def _build_model(self) -> nn.model.Model:
-        model = nn.build_model(self.model_name, self.x.shape[1:], 2, self.agent_name)
+        model = nn.build_model(self.model_name, self.x.shape[1:], 2, None, self.agent_name)
         model.compile(
             tf.keras.optimizers.Adam(self.lr, clipnorm=1.), loss=self.loss, steps_per_execution=100
         )
@@ -405,4 +405,4 @@ def train(self, epoch=40, batch_size=2056):
             self.target_model.set_weights(self.model.get_weights())
 
 
-__all__ = ["DQN"]
\ No newline at end of file
+__all__ = ["DQN"]