Quality fixes

neuralmagic · Nov 4, 2024 · a441dad · a441dad
1 parent 7d93b02
commit a441dad
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/src/guidellm/request/emulated.py b/src/guidellm/request/emulated.py
@@ -405,6 +405,8 @@ def sample_prompt(self, tokens: int) -> str:
 
 
     def sample_images(self):
-        image_indices = self._rng.choice(len(self._images), size=self._config.images, replace=False)
+        image_indices = self._rng.choice(
+            len(self._images), size=self._config.images, replace=False,
+        )
 
         return [self._images[i] for i in image_indices]
diff --git a/src/guidellm/utils/__init__.py b/src/guidellm/utils/__init__.py
@@ -38,4 +38,6 @@
     "resolve_transformers_dataset_split",
     "split_lines_by_punctuation",
     "split_text",
+    "ImageDescriptor",
+    "load_images",
 ]
diff --git a/src/guidellm/utils/images.py b/src/guidellm/utils/images.py
@@ -65,4 +65,4 @@ def load_images(data: str) -> List[ImageDescriptor]:
                     )
                 )
 
-        return images
+    return images
-Original file line number
+Diff line change
@@ Expand Up / @@ -65,4 +65,4 @@ def load_images(data: str) -> List[ImageDescriptor]: @@
                         )
                     )
-            return images
+        return images