Fix imports

adtygan · Jan 30, 2024 · 57949b8 · 57949b8
1 parent 921bcd5
commit 57949b8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/zenml/integrations/hyperai/__init__.py b/src/zenml/integrations/hyperai/__init__.py
@@ -42,7 +42,7 @@ def flavors(cls) -> List[Type[Flavor]]:
         Returns:
             List of stack component flavors for this integration.
         """
-        from zenml.integrations.hyperai.orchestrators import (
+        from zenml.integrations.hyperai.flavors.hyperai_orchestrator_flavor import (
             HyperAIOrchestratorFlavor
         )
 

diff --git a/src/zenml/integrations/hyperai/orchestrators/__init__.py b/src/zenml/integrations/hyperai/orchestrators/__init__.py
@@ -13,9 +13,9 @@
 #  permissions and limitations under the License.
 """HyperAI orchestrator."""
 from zenml.integrations.hyperai.orchestrators.hyperai_orchestrator import (
-    HyperAIOrchestratorFlavor
+    HyperAIOrchestrator
 )
 
 __all__ = [
-    "HyperAIOrchestratorFlavor"
+    "HyperAIOrchestrator"
 ]
diff --git a/tests/integration/integrations/hyperai/orchestrators/test_hyperai_orchestrator.py b/tests/integration/integrations/hyperai/orchestrators/test_hyperai_orchestrator.py
@@ -17,6 +17,8 @@
 from zenml.enums import StackComponentType
 from zenml.integrations.hyperai.orchestrators.hyperai_orchestrator import (
     HyperAIOrchestrator,
+)
+from zenml.integrations.hyperai.flavors.hyperai_orchestrator_flavor import (
     HyperAIOrchestratorConfig,
 )