Skip to content

Commit e7fe1c3

Browse files
committed
move function to get API key to data_validation
* no need to import it from end_of_run_workflow()
1 parent b579996 commit e7fe1c3

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

data_validation.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,15 @@
1+
import os
12
from prefect import task, flow, get_run_logger
23
import time as ttime
34
from tiled.client import from_uri
4-
from end_of_run_workflow import get_api_key_from_env
5+
from dotenv import load_dotenv
6+
7+
8+
def get_api_key_from_env(api_key=None):
9+
with open("/srv/container.secret", "r") as secrets:
10+
load_dotenv(stream=secrets)
11+
api_key = os.environ["TILED_API_KEY"]
12+
return api_key
513

614

715
@task

end_of_run_workflow.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,9 @@
33
from prefect import task, flow, get_run_logger
44
from data_validation import data_validation
55
from test_extra_client import get_other_docs
6-
from dotenv import load_dotenv
76
# from long_flow import long_flow
87

98

10-
def get_api_key_from_env(api_key=None):
11-
with open("/srv/container.secret", "r") as secrets:
12-
load_dotenv(stream=secrets)
13-
api_key = os.environ["TILED_API_KEY"]
14-
return api_key
15-
16-
179
@task
1810
def log_completion(dry_run=False):
1911
logger = get_run_logger()

0 commit comments

Comments
 (0)