lint and formatting

flojoy-ai · Mar 4, 2024 · 221bca5 · 221bca5
1 parent 1a20ddd
commit 221bca5
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/captain/routes/test_sequence.py b/captain/routes/test_sequence.py
@@ -1,4 +1,3 @@
-import asyncio
 from fastapi import APIRouter, WebSocket, WebSocketDisconnect, Depends
 import json
 import pydantic

diff --git a/captain/utils/test_sequencer/generator/generate_test.py b/captain/utils/test_sequencer/generator/generate_test.py
@@ -1,7 +1,6 @@
 import uuid
 from captain.models.test_sequencer import StatusTypes, Test, TestTypes
 from captain.parser.bool_parser.utils.name_validator import validate_name
-from captain.utils.logger import logger
 import os
 from openai import OpenAI
 from captain.utils.test_sequencer.data_stream import _with_error_report

diff --git a/captain/utils/test_sequencer/run_test_sequence.py b/captain/utils/test_sequencer/run_test_sequence.py
@@ -1,4 +1,3 @@
-import traceback
 import time
 from typing import Union, List, Callable
 from flojoy_cloud.client import FlojoyCloudException

diff --git a/pkgs/flojoy/flojoy/_logging.py b/pkgs/flojoy/flojoy/_logging.py
@@ -41,13 +41,17 @@ class StreamEnum(Enum):
 class PipeWriter(Protocol):
     """Protocol for a pipe writer"""
 
-    def write(self, data: bytes): ...
+    def write(self, data: bytes):
+        ...
 
-    def fileno(self) -> int: ...
+    def fileno(self) -> int:
+        ...
 
-    def close(self): ...
+    def close(self):
+        ...
 
-    def flush(self): ...
+    def flush(self):
+        ...
 
 
 # Abstract Pipe class