Merge pull request #9 from JunAishima/fix-beamline-acronym

Fix beamline acronym

Author: JunAishima

data_validation.py

@@ -16,7 +16,7 @@ def get_api_key_from_env(api_key=None):
 
 
 @task(retries=2, retry_delay_seconds=10)
-def get_run(uid, api_key=None, beamline_acronym=None):
+def get_run(uid, api_key=None, beamline_acronym=BEAMLINE_OR_ENDSTATION):
     if not api_key:
         api_key = get_api_key_from_env()
     cl = from_uri("https://tiled.nsls2.bnl.gov", api_key=api_key)

end_of_run_workflow.py

@@ -5,7 +5,7 @@
 from prefect.context import FlowRunContext
 from prefect.settings import PREFECT_UI_URL
 
-from data_validation import data_validation, get_run, get_api_key_from_env
+from data_validation import data_validation, get_run
 
 CATALOG_NAME = "cdi"
 
@@ -33,10 +33,6 @@ def end_of_run_workflow(stop_doc, api_key=None, dry_run=False):
         # Get the uid.
         uid = stop_doc["run_start"]
 
-        # Get Tiled API key, if not set already
-        if not api_key:
-            api_key = get_api_key_from_env()
-
         # Get the scan_id.
         run = get_run(uid, api_key=api_key)
         scan_id = run.start["scan_id"]