fix: resolve lint issues in task stage api

Author: TanyaSingh369-svg

src/pytfe/resources/task_stage.py

@@ -25,14 +25,12 @@ def _parse_task_stage(self, data: dict[str, Any]) -> TaskStage:
         run_data = relationships.get("run", {}).get("data")
         attributes["run"] = run_data
 
-        task_results_data = relationships.get(
-            "task-results", {}
-        ).get("data", [])
+        task_results_data = relationships.get("task-results", {}).get("data", [])
         attributes["task-results"] = task_results_data
 
-        policy_evaluations_data = relationships.get(
-            "policy-evaluations", {}
-        ).get("data", [])
+        policy_evaluations_data = relationships.get("policy-evaluations", {}).get(
+            "data", []
+        )
         attributes["policy-evaluations"] = policy_evaluations_data
 
         return TaskStage.model_validate(attributes)
@@ -80,4 +78,4 @@ def override(
 
         data = response.json().get("data", {})
 
-        return self._parse_task_stage(data)
+        return self._parse_task_stage(data)