@@ -16,21 +16,22 @@ def parameters(input_file):
1616 TTDTarget , TTDInvTarget , TTDLinearTarget
1717 )
1818 with open (input_file , 'r' ) as yaml_file :
19- params = yaml .load (yaml_file )
19+ params = yaml .load (yaml_file , Loader = yaml . SafeLoader )
2020
2121 params ['user_name' ] = getpass .getuser ()
2222 output_path = params ['fs_path' ] + "/" + params ['user_name' ]
2323 base_path = output_path
2424
2525 params ['paths' ]['base_path' ] = base_path
26- params ['paths' ]['signal_prepath' ] = base_path + \
27- params ['paths' ]['signal_prepath' ]
28- params ['paths' ]['shot_list_dir' ] = base_path + \
29- params ['paths' ]['shot_list_dir' ]
26+ params ['paths' ]['signal_prepath' ] = (
27+ base_path + params ['paths' ]['signal_prepath' ])
28+ params ['paths' ]['shot_list_dir' ] = (
29+ base_path + params ['paths' ]['shot_list_dir' ])
3030 params ['paths' ]['output_path' ] = output_path
3131 h = get_unique_signal_hash (sig .all_signals .values ())
32- params ['paths' ]['global_normalizer_path' ] = output_path + \
33- '/normalization/normalization_signal_group_{}.npz' .format (h )
32+ params ['paths' ]['global_normalizer_path' ] = (
33+ output_path
34+ + '/normalization/normalization_signal_group_{}.npz' .format (h ))
3435 if params ['training' ]['hyperparam_tuning' ]:
3536 # params['paths']['saved_shotlist_path'] =
3637 # './normalization/shot_lists.npz'
@@ -44,12 +45,12 @@ def parameters(input_file):
4445 # '/normalization/shot_lists.npz'
4546 params ['paths' ]['normalizer_path' ] = (
4647 params ['paths' ]['global_normalizer_path' ])
47- params ['paths' ]['model_save_path' ] = (
48- output_path + '/model_checkpoints/' )
48+ params ['paths' ]['model_save_path' ] = (output_path
49+ + '/model_checkpoints/' )
4950 params ['paths' ]['csvlog_save_path' ] = output_path + '/csv_logs/'
5051 params ['paths' ]['results_prepath' ] = output_path + '/results/'
51- params ['paths' ]['tensorboard_save_path' ] = output_path + \
52- params ['paths' ]['tensorboard_save_path' ]
52+ params ['paths' ]['tensorboard_save_path' ] = (
53+ output_path + params ['paths' ]['tensorboard_save_path' ])
5354 params ['paths' ]['saved_shotlist_path' ] = (
5455 params ['paths' ]['base_path' ] + '/processed_shotlists/'
5556 + params ['paths' ]['data' ]
0 commit comments