Skip to content

Commit a441dad

Browse files
committed
Quality fixes
1 parent 7d93b02 commit a441dad

File tree

3 files changed

+6
-2
lines changed

3 files changed

+6
-2
lines changed

src/guidellm/request/emulated.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,8 @@ def sample_prompt(self, tokens: int) -> str:
405405

406406

407407
def sample_images(self):
408-
image_indices = self._rng.choice(len(self._images), size=self._config.images, replace=False)
408+
image_indices = self._rng.choice(
409+
len(self._images), size=self._config.images, replace=False,
410+
)
409411

410412
return [self._images[i] for i in image_indices]

src/guidellm/utils/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,6 @@
3838
"resolve_transformers_dataset_split",
3939
"split_lines_by_punctuation",
4040
"split_text",
41+
"ImageDescriptor",
42+
"load_images",
4143
]

src/guidellm/utils/images.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,4 +65,4 @@ def load_images(data: str) -> List[ImageDescriptor]:
6565
)
6666
)
6767

68-
return images
68+
return images

0 commit comments

Comments
 (0)