Skip to content

Commit

Permalink
paths
Browse files Browse the repository at this point in the history
  • Loading branch information
joglekara committed Nov 13, 2024
1 parent 5f9696f commit 3798f99
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions tsadar/process/prepare.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,10 +25,10 @@ def prepare_data(config: Dict) -> Dict:
custom_path = None
if "filenames" in config["data"].keys():
if config["data"]["filenames"]["epw"] is not None:
custom_path = os.path.dirname(config["data"]["filenames"]["tempepwfilepath"])
custom_path = os.path.dirname(config["data"]["filenames"]["epw-local"])

if config["data"]["filenames"]["iaw"] is not None:
custom_path = os.path.dirname(config["data"]["filenames"]["tempiawfilepath"])
custom_path = os.path.dirname(config["data"]["filenames"]["iaw-local"])

[elecData, ionData, xlab, config["other"]["extraoptions"]["spectype"]] = loadData(
config["data"]["shotnum"], config["data"]["shotDay"], config["other"]["extraoptions"], custom_path=custom_path
Expand Down
4 changes: 2 additions & 2 deletions tsadar/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,12 +91,12 @@ def run_for_app(run_id: str) -> str:
config = yaml.safe_load(fi)

if config["data"]["filenames"]["epw"] is not None:
config["data"]["filenames"]["tempepwfilepath"] = utils.download_file(
config["data"]["filenames"]["epw-local"] = utils.download_file(
config["data"]["filenames"]["epw"], mlflow_run.info.artifact_uri, temp_path
)

if config["data"]["filenames"]["iaw"] is not None:
config["data"]["filenames"]["tempiawfilepath"] = utils.download_file(
config["data"]["filenames"]["iaw-local"] = utils.download_file(
config["data"]["filenames"]["iaw"], mlflow_run.info.artifact_uri, temp_path
)

Expand Down

0 comments on commit 3798f99

Please sign in to comment.