[BugFix] fix inheritance structure in l2d

ai4co · Jun 3, 2024 · 6744ddb · 6744ddb
1 parent 84f39fb
commit 6744ddb
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,