Skip to content

Commit

Permalink
Fix paths in simulation training test
Browse files Browse the repository at this point in the history
  • Loading branch information
roman-bushuiev committed Feb 10, 2025
1 parent c925240 commit a5e89df
Showing 1 changed file with 3 additions and 8 deletions.
11 changes: 3 additions & 8 deletions tests/test_simulation_baseline_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,17 +16,13 @@ def simulation_baseline_training():
from massspecgym.models.simulation.prec_only import PrecOnlySimulationMassSpecGymModel
from massspecgym.simulation_utils.run_utils import load_config, get_split_ss



# default settings for simulation runs
template_fp = "../config/simulation/template.yml"
template_fp = "config/simulation/template.yml"
# special settings for this demo
custom_fp = "../config/simulation/demo.yml"
custom_fp = "config/simulation/demo.yml"

# W&B logging is disabled
wandb_mode = "disabled"
# directory for saving model checkpoints
checkpoint_dp = "../checkpoint/simulation"

config_d = load_config(template_fp,custom_fp)

Expand Down Expand Up @@ -58,8 +54,7 @@ def simulation_baseline_training():
project=config_d["wandb_project"],
entity=config_d["wandb_entity"],
name=config_d["wandb_name"],
mode=wandb_mode,
dir=checkpoint_dp,
mode=wandb_mode
)
logger = pl.loggers.WandbLogger(
entity=config_d["wandb_entity"],
Expand Down

0 comments on commit a5e89df

Please sign in to comment.