clean up typing

mosaicml · Aug 17, 2023 · 56aa68c · 56aa68c
1 parent a61af04
commit 56aa68c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llmfoundry/models/mosaicbert/configuration_mosaicbert.py b/llmfoundry/models/mosaicbert/configuration_mosaicbert.py
@@ -5,6 +5,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from transformers import BertConfig as TransformersBertConfig
+from typing import Any
 
 
 class BertConfig(TransformersBertConfig):
@@ -14,7 +15,7 @@ def __init__(
         self,
         alibi_starting_size: int = 512,
         attention_probs_dropout_prob: float = 0.0,
-        **kwargs,
+        **kwargs: Any,
     ):
         """Configuration class for MosaicBert.