Fix typo (#966)

mosaicml · Feb 9, 2024 · c7c9d24 · c7c9d24
1 parent aa0ea6e
commit c7c9d24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/utils/warnings.py b/llmfoundry/utils/warnings.py
@@ -14,7 +14,7 @@ class VersionedDeprecationWarning(DeprecationWarning):
         ...     warnings.warn(
         ...         VersionedDeprecationWarning(
         ...             "Function XYZ is deprecated.",
-        ...             after_version="2.0.0"
+        ...             remove_version="2.0.0"
         ...         )
         ...     )
         ...