refactor code C51

french-ai · Dec 7, 2020 · 3619ee6 · 3619ee6
1 parent 935e493
commit 3619ee6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):