salmon-refrigerator-32115
02/10/2023, 5:12 PMglamorous-carpet-83516
02/10/2023, 8:40 PMsalmon-refrigerator-32115
02/10/2023, 9:25 PMsalmon-refrigerator-32115
02/10/2023, 10:17 PMglamorous-carpet-83516
02/10/2023, 10:52 PMsalmon-refrigerator-32115
02/11/2023, 12:17 AMimport os
from flytekit import task, workflow
@task
def get_domain() -> str:
domain = os.environ["FLYTE_INTERNAL_TASK_DOMAIN"]
print(f'FLYTE_INTERNAL_TASK_DOMAIN: {domain}')
return domain
@task
def get_env_var() -> str:
my_var = os.environ["ML_S3_BUCKET_PREFIX"]
print(f'ML_S3_BUCKET_PREFIX: {my_var}')
return my_var
@workflow
def wf() -> None:
get_env_var()
get_domain()
salmon-refrigerator-32115
02/11/2023, 12:17 AMsalmon-refrigerator-32115
02/11/2023, 12:18 AMglamorous-carpet-83516
02/11/2023, 12:39 AMdomain = os.getenv("FLYTE_INTERNAL_TASK_DOMAIN", "default_value")
glamorous-carpet-83516
02/11/2023, 12:39 AMdef get_domain() -> Optional[str]
salmon-refrigerator-32115
02/11/2023, 12:59 AM