maybe fix docstring

mosaicml · Mar 13, 2024 · 88700f4 · 88700f4
1 parent 006b63a
commit 88700f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/composer/utils/warnings.py b/composer/utils/warnings.py
@@ -14,6 +14,7 @@ class VersionedDeprecationWarning(DeprecationWarning):
         remove_version (str): The version in which the feature will be removed.
 
     Example:
+        >>> import warnings
         >>> def deprecated_function():
         ...     warnings.warn(
         ...         VersionedDeprecationWarning(