chore: minor docstring changes | [#197]

AntonioMrtz · Aug 14, 2024 · 0b11aa2 · 0b11aa2
1 parent f2ecfc8
commit 0b11aa2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Backend/app/database/DatabaseTestingConnection.py b/Backend/app/database/DatabaseTestingConnection.py
@@ -15,7 +15,7 @@ def _get_mongo_client(self):  # noqa: ANN202
         """Get mock Mongo client class
 
         Returns:
-            Any: the mock Mongo client class
+            MongoClientMock: the mock Mongo client class
         """
         enable_gridfs_integration()
         return MongoClientMock

diff --git a/Backend/app/database/database_schema.py b/Backend/app/database/database_schema.py
@@ -31,6 +31,8 @@ class DatabaseCollection(StrEnum):
 
 
 def __is_connection__init__(func: Callable):  # noqa: ANN202
+    """Check database connection is initialized"""
+
     @wraps(func)
     def wrapper(*args, **kwargs):  # noqa: ANN202
         if BaseDatabaseConnection.connection is not None: