From 56aa68ce9e374d070b67fe41d7012e9d8a99be4c Mon Sep 17 00:00:00 2001
From: jacobfulano <jportes@me.com>
Date: Thu, 17 Aug 2023 13:15:00 -0400
Subject: [PATCH] clean up typing

---
 llmfoundry/models/mosaicbert/configuration_mosaicbert.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/llmfoundry/models/mosaicbert/configuration_mosaicbert.py b/llmfoundry/models/mosaicbert/configuration_mosaicbert.py
index 048dd39451..94175cff59 100644
--- 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.