Use RegistryBundleBase instead of RegistryBundle

Summary: As titled Reviewed By: saitcakmak Differential Revision: D67423166
facebook · Dec 19, 2024 · b855580 · b855580
1 parent a3da9a5
commit b855580
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ax/preview/api/configs.py b/ax/preview/api/configs.py
@@ -10,7 +10,7 @@
 from typing import Any, Callable, List, Mapping, Sequence
 
 from ax.preview.api.types import TParameterValue
-from ax.storage.registry_bundle import RegistryBundle
+from ax.storage.registry_bundle import RegistryBundleBase
 
 
 class ParameterType(Enum):
@@ -163,4 +163,4 @@ class OrchestrationConfig:
 class StorageConfig:
     creator: Callable[..., Any] | None = None  # pyre-fixme[4]
     url: str | None = None
-    registry_bundle: RegistryBundle | None = None
+    registry_bundle: RegistryBundleBase | None = None