Fix issue with CE scheduler deployment and management CLI

polyaxon · Feb 12, 2024 · fe44462 · fe44462
1 parent d351223
commit fe44462
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/haupt/haupt/cli/runners/manage.py b/haupt/haupt/cli/runners/manage.py
@@ -44,7 +44,7 @@ def run_manage(command: str, args: Optional[List[str]] = None):
     from polyaxon._env_vars.keys import ENV_KEYS_SERVICE, ENV_KEYS_UI_IN_SANDBOX
     from polyaxon._services.values import PolyaxonServices
 
-    os.environ[ENV_KEYS_SERVICE] = PolyaxonServices.API
-    os.environ[ENV_KEYS_UI_IN_SANDBOX] = "true"
-    settings.set_sandbox_config()
+    if os.environ[ENV_KEYS_SERVICE] == PolyaxonServices.API:
+        os.environ[ENV_KEYS_UI_IN_SANDBOX] = "true"
+        settings.set_sandbox_config()
     manage(command, args)
diff --git a/haupt/haupt/polyconf/urls.py b/haupt/haupt/polyconf/urls.py
@@ -1,6 +1,8 @@
 from haupt.polyconf.settings import PLATFORM_CONFIG
 
-if PLATFORM_CONFIG.is_streams_service:
+if PLATFORM_CONFIG.is_scheduler_service:
+    pass
+elif PLATFORM_CONFIG.is_streams_service:
     from haupt.streams.patterns import *  # noqa
 else:
     from haupt.apis.patterns import *  # noqa