Skip to content

Commit

Permalink
remove rich
Browse files Browse the repository at this point in the history
  • Loading branch information
aniketmaurya committed Dec 10, 2024
1 parent 808a4ac commit 7aff112
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions src/litserve/loops.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@
from typing import Any, Dict, List, Optional, Tuple, Union

from fastapi import HTTPException
from rich.progress import track
from starlette.formparsers import MultiPartParser

from litserve import LitAPI
Expand Down Expand Up @@ -697,10 +696,6 @@ def step(
self.active_sequences.clear()
return responses

def request_finished(self, lit_api: LitAPI, lit_spec: Optional[LitSpec]) -> bool:
"""Check if all sequences are processed."""
return len(self.active_sequences) == 0

def prefill(
self,
pending_requests: List[Tuple[str, Any]],
Expand Down Expand Up @@ -777,7 +772,7 @@ def run(
prev_outputs = responses

# Send responses for all sequences (both streaming and completed)
for step_output in track(responses, description="Processing responses"):
for step_output in responses:
logger.debug(f"Processing response: {step_output}")
status = step_output.status
response_data = step_output.output
Expand Down

0 comments on commit 7aff112

Please sign in to comment.