DH-4724/reformat with black

Dataherald · Sep 27, 2023 · a7b86e7 · a7b86e7
1 parent 14ed35e
commit a7b86e7
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/dataherald/db/__init__.py b/dataherald/db/__init__.py
@@ -29,7 +29,14 @@ def find_by_id(self, collection: str, id: str) -> dict:
         pass
 
     @abstractmethod
-    def find(self, collection: str, query: dict, sort: list = None, page: int = 0, limit: int = 0) -> list:
+    def find(
+        self,
+        collection: str,
+        query: dict,
+        sort: list = None,
+        page: int = 0,
+        limit: int = 0,
+    ) -> list:
         pass
 
     @abstractmethod

diff --git a/dataherald/db/mongo.py b/dataherald/db/mongo.py
@@ -40,7 +40,14 @@ def find_by_id(self, collection: str, id: str) -> dict:
         return self._data_store[collection].find_one({"_id": ObjectId(id)})
 
     @override
-    def find(self, collection: str, query: dict, sort: list = None, page: int = 0, limit: int = 0) -> list:
+    def find(
+        self,
+        collection: str,
+        query: dict,
+        sort: list = None,
+        page: int = 0,
+        limit: int = 0,
+    ) -> list:
         skip_count = (page - 1) * limit
         cursor = self._data_store[collection].find(query)
         if sort:

diff --git a/dataherald/tests/db/test_db.py b/dataherald/tests/db/test_db.py
@@ -55,7 +55,14 @@ def find_by_id(self, collection: str, id: str) -> dict:
         return None
 
     @override
-    def find(self, collection: str, query: dict, sort: list = None, page: int = 0, limit: int = 0) -> list:
+    def find(
+        self,
+        collection: str,
+        query: dict,
+        sort: list = None,
+        page: int = 0,
+        limit: int = 0,
+    ) -> list:
         return []
 
     @override