From 789a56bf63ba61d1ac819b7b61dc1c9d4cf56834 Mon Sep 17 00:00:00 2001 From: saileshwar-skyflow Date: Thu, 13 Mar 2025 15:11:47 +0530 Subject: [PATCH] SK-1772: Update enums --- skyflow/utils/_utils.py | 2 +- skyflow/utils/enums/env.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/skyflow/utils/_utils.py b/skyflow/utils/_utils.py index 5002956a..e0d8783d 100644 --- a/skyflow/utils/_utils.py +++ b/skyflow/utils/_utils.py @@ -61,7 +61,7 @@ def get_vault_url(cluster_id, env,vault_id, logger = None): raise SkyflowError(SkyflowMessages.Error.INVALID_ENV.value.format(vault_id), invalid_input_error_code) base_url = EnvUrls[env.name].value - protocol = "https" if env != Env.PROD else "http" + protocol = "https" return f"{protocol}://{cluster_id}.{base_url}" diff --git a/skyflow/utils/enums/env.py b/skyflow/utils/enums/env.py index 862f8f8a..1f2f7f17 100644 --- a/skyflow/utils/enums/env.py +++ b/skyflow/utils/enums/env.py @@ -1,13 +1,13 @@ from enum import Enum class Env(Enum): - DEV = 'DEV', - SANDBOX = 'SANDBOX', + DEV = 'DEV' + SANDBOX = 'SANDBOX' PROD = 'PROD' STAGE = 'STAGE' class EnvUrls(Enum): - PROD = "vault.skyflowapis.com", - SANDBOX = "vault.skyflowapis-preview.com", + PROD = "vault.skyflowapis.com" + SANDBOX = "vault.skyflowapis-preview.com" DEV = "vault.skyflowapis.dev" STAGE = "vault.skyflowapis.tech" \ No newline at end of file