ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Hydra should not store the full resolved OmegaConf (issue #327)

Browse Source
This commit is contained in:
allegroai 2021-05-19 15:25:47 +03:00
parent ed75eac6ab
commit 65b838f409
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clearml/binding/hydra_bind.py
View File

@ -113,7 +113,7 @@ class PatchHydra(object):
pre_app_task_init_call = bool(PatchHydra._current_task) pre_app_task_init_call = bool(PatchHydra._current_task)
if pre_app_task_init_call: if pre_app_task_init_call and not running_remotely():
LoggerRoot.get_base_logger(PatchHydra).info( LoggerRoot.get_base_logger(PatchHydra).info(
'Task.init called outside of Hydra-App. For full Hydra multi-run support, ' 'Task.init called outside of Hydra-App. For full Hydra multi-run support, '
'move the Task.init call into the Hydra-App main function') 'move the Task.init call into the Hydra-App main function')
@ -158,11 +158,15 @@ class PatchHydra(object):
description = 'Full OmegaConf YAML configuration overridden! ({}/{}=True)'.format( description = 'Full OmegaConf YAML configuration overridden! ({}/{}=True)'.format(
PatchHydra._parameter_section, PatchHydra._parameter_allow_full_edit) PatchHydra._parameter_section, PatchHydra._parameter_allow_full_edit)
# we should not have the hydra section in the config, but this seems never to be the case anymore.
# config = config.copy()
# config.pop('hydra', None)
configuration = dict( configuration = dict(
name=PatchHydra._config_section, name=PatchHydra._config_section,
description=description, description=description,
config_type='OmegaConf YAML', config_type='OmegaConf YAML',
config_text=OmegaConf.to_yaml({k: v for k, v in config.items() if k not in ('hydra', )}) config_text=OmegaConf.to_yaml(config, resolve=False)
) )
if PatchHydra._current_task: if PatchHydra._current_task:
# noinspection PyProtectedMember # noinspection PyProtectedMember