From 3619ee6c624ebff1945e2e5a681e2129a9e349e8 Mon Sep 17 00:00:00 2001
From: bruzat <nathan.bruzat@viacesi.fr>
Date: Mon, 7 Dec 2020 09:41:45 +0100
Subject: [PATCH] refactor code C51

---
 blobrl/networks/c51_network.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/blobrl/networks/c51_network.py b/blobrl/networks/c51_network.py
index 6f13211..a82b4ec 100644
--- a/blobrl/networks/c51_network.py
+++ b/blobrl/networks/c51_network.py
@@ -67,7 +67,7 @@ def forward(self, observation):
             q = q.permute(1, 0, 2)
 
             return q
-        elif isinstance(self.action_space, MultiDiscrete):
+        if isinstance(self.action_space, MultiDiscrete):
 
             def do_forward(nvec, llayers, x):
                 if isinstance(llayers[-1], list):