diff --git a/utils/callbacks.py b/utils/callbacks.py
index c6b3be1cbd69..2b32df0bf1c1 100644
--- a/utils/callbacks.py
+++ b/utils/callbacks.py
@@ -38,9 +38,9 @@ def register_action(self, hook, name='', callback=None):
         Register a new action to a callback hook
 
         Args:
-            hook        The callback hook name to register the action to
-            name        The name of the action for later reference
-            callback    The callback to fire
+            hook: The callback hook name to register the action to
+            name: The name of the action for later reference
+            callback: The callback to fire
         """
         assert hook in self._callbacks, f"hook '{hook}' not found in callbacks {self._callbacks}"
         assert callable(callback), f"callback '{callback}' is not callable"
@@ -51,21 +51,18 @@ def get_registered_actions(self, hook=None):
         Returns all the registered actions by callback hook
 
         Args:
-            hook The name of the hook to check, defaults to all
+            hook: The name of the hook to check, defaults to all
         """
-        if hook:
-            return self._callbacks[hook]
-        else:
-            return self._callbacks
+        return self._callbacks[hook] if hook else self._callbacks
 
     def run(self, hook, *args, **kwargs):
         """
         Loop through the registered actions and fire all callbacks
 
         Args:
-            hook The name of the hook to check, defaults to all
-            args Arguments to receive from YOLOv5
-            kwargs Keyword Arguments to receive from YOLOv5
+            hook: The name of the hook to check, defaults to all
+            args: Arguments to receive from YOLOv5
+            kwargs: Keyword Arguments to receive from YOLOv5
         """
 
         assert hook in self._callbacks, f"hook '{hook}' not found in callbacks {self._callbacks}"