Skip to content

Commit 3a3ea87

Browse files
committed
Implementation and tests for the core utils packages for the scheduler refactor
1 parent cff8d91 commit 3a3ea87

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

src/guidellm/utils/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,9 @@
7878
"RunningStats",
7979
"SerializationTypesAlias",
8080
"Serializer",
81+
"PydanticClassRegistryMixin",
82+
"RegistryMixin",
83+
"ReloadableBaseModel",
8184
"SingletonMixin",
8285
"StandardBaseDict",
8386
"StandardBaseModel",
@@ -86,6 +89,7 @@
8689
"ThreadSafeSingletonMixin",
8790
"TimeRunningStats",
8891
"all_defined",
92+
"ThreadSafeSingletonMixin",
8993
"check_load_processor",
9094
"clean_text",
9195
"filter_text",

tests/unit/utils/test_registry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -531,3 +531,4 @@ def walk_packages(package_path, package_name):
531531
assert len(objects) == 1
532532
assert TestAutoRegistry.registry_populated is True
533533
assert TestAutoRegistry.registry is not None
534+
assert "module1class" in TestAutoRegistry.registry

0 commit comments

Comments
 (0)