From 341ef458b440e98158c8038f20f873c7ce0bbd0e Mon Sep 17 00:00:00 2001
From: Alexander Goryunov <alex.goryunov@gmail.com>
Date: Mon, 10 Jul 2023 20:48:32 +0300
Subject: [PATCH] type hints

---
 torchscale/architecture/encoder_decoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchscale/architecture/encoder_decoder.py b/torchscale/architecture/encoder_decoder.py
index d91313f..ed64641 100644
--- a/torchscale/architecture/encoder_decoder.py
+++ b/torchscale/architecture/encoder_decoder.py
@@ -11,7 +11,7 @@ from torchscale.architecture.encoder import Encoder
 class EncoderDecoder(nn.Module):
     def __init__(
         self,
-        args,
+        args: EncoderDecoderConfig,
         encoder_embed_tokens=None,
         encoder_embed_positions=None,
         decoder_embed_tokens=None,