DH-4724/reformat with black

dataherald/api/__init__.py

@@ -102,9 +102,7 @@ def get_golden_records(self, page: int = 1, limit: int = 10) -> List[GoldenRecor
         pass
 
     @abstractmethod
-    def add_instruction(
-        self, instruction_request: InstructionRequest
-    ) -> Instruction:
+    def add_instruction(self, instruction_request: InstructionRequest) -> Instruction:
         pass
 
     @abstractmethod

dataherald/api/fastapi.py

@@ -318,9 +318,7 @@ def get_golden_records(self, page: int = 1, limit: int = 10) -> List[GoldenRecor
         return all_records[start_idx:end_idx]
 
     @override
-    def add_instruction(
-        self, instruction_request: InstructionRequest
-    ) -> Instruction:
+    def add_instruction(self, instruction_request: InstructionRequest) -> Instruction:
         instruction_repository = InstructionRepository(self.storage)
         instruction = Instruction(
             instruction=instruction_request.instruction,
@@ -342,7 +340,7 @@ def get_instructions(
         return instruction_repository.find_all()
 
     @override
-    def delete_instruction(self,instruction_id: str) -> dict:
+    def delete_instruction(self, instruction_id: str) -> dict:
         instruction_repository = InstructionRepository(self.storage)
         deleted = instruction_repository.delete_by_id(instruction_id)
         if deleted == 0:

dataherald/server/fastapi/__init__.py

@@ -261,13 +261,9 @@ def get_golden_records(self, page: int = 1, limit: int = 10) -> List[GoldenRecor
         """Gets golden records"""
         return self._api.get_golden_records(page, limit)
 
-    def add_instruction(
-        self, instruction_request: InstructionRequest
-    ) -> Instruction:
+    def add_instruction(self, instruction_request: InstructionRequest) -> Instruction:
         """Adds an instruction"""
-        created_records = self._api.add_instruction(
-            instruction_request
-        )
+        created_records = self._api.add_instruction(instruction_request)
 
         # Return a JSONResponse with status code 201 and the location header.
         instruction_as_dict = created_records.dict()
@@ -292,6 +288,4 @@ def update_instruction(
         instruction_request: UpdateInstruction,
     ) -> Instruction:
         """Updates an instruction"""
-        return self._api.update_instruction(
-            instruction_id, instruction_request
-        )
+        return self._api.update_instruction(instruction_id, instruction_request)

dataherald/types.py

@@ -39,9 +39,11 @@ class NLQuery(BaseModel):
     question: str
     db_connection_id: str
 
+
 class UpdateInstruction(BaseModel):
     instruction: str
 
+
 class InstructionRequest(DBConnectionValidation):
     instruction: str