lint

mosaicml · Jun 25, 2024 · b74330e · b74330e
1 parent 0a3f1b4
commit b74330e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/models/mpt/configuration_mpt.py b/llmfoundry/models/mpt/configuration_mpt.py
@@ -211,7 +211,7 @@ def __init__(
 
         self._validate_config()
 
-    def _validate_block_overrides(self, block_overrides):
+    def _validate_block_overrides(self, block_overrides: Dict[str, Any]):
         warnings.warn(ExperimentalWarning('block_overrides'))
         if 'start' not in block_overrides and 'repeating_pattern' not in block_overrides and 'end' not in block_overrides:
             raise ValueError(