[BugFix] fix inheritance structure in L2D decoder

ai4co · Jun 3, 2024 · 6b28168 · 6b28168
1 parent f4abe1b
commit 6b28168
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rl4co/models/zoo/l2d/decoder.py b/rl4co/models/zoo/l2d/decoder.py
@@ -313,7 +313,7 @@ def pre_decoder_hook(
         return td, env, (cache,)
 
 
-class L2DAttnActor(AttentionModelDecoder):
+class L2DAttnActor(AttnActor):
     def __init__(
         self,
         embed_dim: int = 128,