Skip to content

Commit 54138aa

Browse files
committed
fixed precommit, lint/mypy errrors
1 parent 7f35220 commit 54138aa

File tree

2 files changed

+105
-106
lines changed

2 files changed

+105
-106
lines changed

src/guidellm/dataset/synthetic.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44
from pathlib import Path
55
from typing import Any, Literal, Optional, Union
66

7-
import random
8-
import time
97
import yaml
108
from datasets import (
119
Dataset,
@@ -177,12 +175,16 @@ def __iter__(
177175
# Increment counter for each request
178176
self.request_counter += 1
179177
yield {
180-
"prompt": self._create_prompt(prompt_tokens, start_index, self.request_counter),
178+
"prompt": self._create_prompt(
179+
prompt_tokens, start_index, self.request_counter
180+
),
181181
"prompt_tokens_count": prompt_tokens,
182182
"output_tokens_count": output_tokens,
183183
}
184184

185-
def _create_prompt(self, prompt_tokens: int, start_index: int, request_id: int) -> str:
185+
def _create_prompt(
186+
self, prompt_tokens: int, start_index: int, request_id: int
187+
) -> str:
186188
"""
187189
Create a prompt with unique prefix to prevent vLLM prefix caching.
188190
Args:

0 commit comments

Comments
 (0)