fix unused variable

microsoft · Apr 24, 2024 · 82016f1 · 82016f1
1 parent eac26b2
commit 82016f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/olive/passes/__init__.py b/olive/passes/__init__.py
@@ -12,4 +12,5 @@
     "PassParamDefault",
     "PassModuleConfig",
     "FullPassConfig",
+    "REGISTRY",
 ]
diff --git a/olive/strategy/search_algorithm/__init__.py b/olive/strategy/search_algorithm/__init__.py
@@ -9,4 +9,4 @@
 
 REGISTRY = SearchAlgorithm.registry
 
-__all__ = ["SearchAlgorithm", "ExhaustiveSearchAlgorithm", "RandomSearchAlgorithm", "TPESearchAlgorithm"]
+__all__ = ["SearchAlgorithm", "ExhaustiveSearchAlgorithm", "RandomSearchAlgorithm", "TPESearchAlgorithm", "REGISTRY"]