code quality

mosaicml · Mar 13, 2024 · 006b63a · 006b63a
1 parent fb20988
commit 006b63a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/composer/core/time.py b/composer/core/time.py
@@ -547,7 +547,7 @@ def get_state(self) -> Dict[str, Union[Time[int], datetime.timedelta]]:
             Dict[str, Union[Time[int], datetime.timedelta]]: All values of the timestamp object.
         """
         warnings.warn(
-            VersionedDeprecationWarning('core.time.Timestamp.get_state is deprecated.', remove_version='0.21.0')
+            VersionedDeprecationWarning('core.time.Timestamp.get_state is deprecated.', remove_version='0.21.0'),
         )
 
         return self.state_dict()

diff --git a/tests/utils/test_warnings.py b/tests/utils/test_warnings.py
@@ -14,6 +14,7 @@ def deprecated_function():
         warnings.warn(VersionedDeprecationWarning('This function is deprecated.', remove_version='0.20.0'))
 
     with pytest.warns(
-        VersionedDeprecationWarning, match='This function is deprecated. It will be removed in version 0.20.0.'
+        VersionedDeprecationWarning,
+        match='This function is deprecated. It will be removed in version 0.20.0.',
     ):
         deprecated_function()