Skip to content
Closed
14 changes: 8 additions & 6 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ keywords = [
]
dependencies = [
"click>=8.0.0,<8.2.0",
"culsans~=0.9.0",
"datasets",
"ftfy>=6.0.0",
"httpx[http2]<1.0.0",
Expand Down Expand Up @@ -157,15 +158,16 @@ indent-style = "space"

[tool.ruff.lint]
ignore = [
"PLR0913",
"TC001",
"COM812",
"ISC001",
"TC002",
"COM812", # ignore trailing comma errors due to older Python versions
"PD011", # ignore .values usage since ruff assumes it's a Pandas DataFrame
"PLR0913", # ignore too many arguments in function definitions
"PLW1514", # allow Path.open without encoding
"RET505", # allow `else` blocks
"RET506", # allow `else` blocks
"PD011", # ignore .values usage since ruff assumes it's a Pandas DataFrame
"S311", # allow standard pseudo-random generators
"TC001", # ignore imports used only for type checking
"TC002", # ignore imports used only for type checking
"TC003", # ignore imports used only for type checking
]
select = [
# Rules reference: https://docs.astral.sh/ruff/rules/
Expand Down
22 changes: 22 additions & 0 deletions src/guidellm/utils/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,13 @@
from .auto_importer import AutoImporterMixin
from .colors import Colors
from .default_group import DefaultGroupHandler
from .encoding import (
Encoder,
EncodingTypesAlias,
MessageEncoding,
SerializationTypesAlias,
Serializer,
)
from .functions import (
all_defined,
safe_add,
Expand All @@ -16,6 +23,12 @@
from .hf_transformers import (
check_load_processor,
)
from .messaging import (
InterProcessMessaging,
InterProcessMessagingManagerQueue,
InterProcessMessagingPipe,
InterProcessMessagingQueue,
)
from .pydantic_utils import (
PydanticClassRegistryMixin,
ReloadableBaseModel,
Expand Down Expand Up @@ -49,13 +62,22 @@
"Colors",
"DefaultGroupHandler",
"DistributionSummary",
"Encoder",
"EncodingTypesAlias",
"EndlessTextCreator",
"IntegerRangeSampler",
"InterProcessMessaging",
"InterProcessMessagingManagerQueue",
"InterProcessMessagingPipe",
"InterProcessMessagingQueue",
"MessageEncoding",
"Percentiles",
"PydanticClassRegistryMixin",
"RegistryMixin",
"ReloadableBaseModel",
"RunningStats",
"SerializationTypesAlias",
"Serializer",
"SingletonMixin",
"StandardBaseDict",
"StandardBaseModel",
Expand Down
Loading
Loading