diff --git a/tests/test_trainer_utils.py b/tests/test_trainer_utils.py
index f03f62f3e..beed009a5 100644
--- a/tests/test_trainer_utils.py
+++ b/tests/test_trainer_utils.py
@@ -1,3 +1,4 @@
+"""Module containing tests for trainer utility functions."""
 import unittest
 from functools import partial