def get_calib_manager(args, unknownArgs, force_metadata=False):
mod = args.loaded_module
if not hasattr(
mod,
'run_calib_args') or 'calib_manager' not in mod.run_calib_args:
warning_note = \
"""
/!\\ WARNING /!\\ Suggest to set calib_manager within run_calib_args like the following:
run_calib_args = {'calib_manager': calib_manager}
"""
print(warning_note)
manager = mod.calib_manager
else:
manager = mod.run_calib_args['calib_manager']
# Atiye: uncomment this block if quiet is not being honored down below in the code. If it isn't we should fix
# that anyway. This message will not self destruct, ever.
# try:
# if args.quiet:
# mod.run_sim_args['quiet'] = True
# except:
# pass
# Update the SetupParser to match the existing experiment environment/block if force_metadata == True
if force_metadata:
exp = manager.get_experiment_from_iteration(
iteration=args.iteration, force_metadata=force_metadata)
SetupParser.override_block(exp.selected_block)
return manager
def test_explicit_overrides(self):
filename = os.path.join(self.input_path, SetupParser.ini_filename)
overrides = {'in_all': 'yellow'}
SetupParser.init(selected_block='TEST',
setup_file=filename,
overrides=overrides,
is_testing=True)
self.assertEqual(SetupParser.get('in_all'), overrides['in_all'])
# verify this works even after block change
SetupParser.override_block(block='LOCAL2')
self.assertEqual(SetupParser.get('in_all'), overrides['in_all'])
def test_verify_override_block(self):
"""
1) set the SetupParser once to get the expected value, unset it, then 2) load a
different selected block & then override it to the first. Should get same answer
both routes.
"""
SetupParser.init(selected_block='LOCAL', is_testing=True)
expected_params = dict(SetupParser.items(SetupParser.selected_block))
SetupParser._uninit()
SetupParser.init(selected_block='HPC', is_testing=True)
SetupParser.override_block('LOCAL')
override_params = dict(SetupParser.items(SetupParser.selected_block))
self.assertEqual(override_params, expected_params)
def get_calib_manager(args, unknownArgs, force_metadata=False):
mod = args.loaded_module
if not hasattr(
mod,
'run_calib_args') or 'calib_manager' not in mod.run_calib_args:
warning_note = \
"""
/!\\ WARNING /!\\ Suggest to set calib_manager within run_calib_args like the following:
run_calib_args = {'calib_manager': calib_manager}
"""
print(warning_note)
manager = mod.calib_manager
else:
manager = mod.run_calib_args['calib_manager']
# Update the SetupParser to match the existing experiment environment/block if force_metadata == True
if force_metadata:
exp = manager.get_experiment_from_iteration(
iteration=args.iteration, force_metadata=force_metadata)
SetupParser.override_block(exp.selected_block)
return manager
def get_resamplers(args, unknownArgs, force_metadata=False):
mod = args.loaded_module
if not hasattr(mod,
'run_calib_args') or 'resamplers' not in mod.run_calib_args:
warning_note = \
"""
/!\\ WARNING /!\\ Required to set resamplers within run_calib_args like the following:
run_calib_args = {'resamplers': [xxx_resampler1, yyy_resampler2, ...]}
"""
print(warning_note)
exit()
else:
resamplers = mod.run_calib_args['resamplers']
# Update the SetupParser to match the existing experiment environment/block if force_metadata == True
if force_metadata:
exp = resamplers[0].calib_manager.get_experiment_from_iteration(
iteration=args.iteration, force_metadata=force_metadata)
SetupParser.override_block(exp.selected_block)
return resamplers
