diff --git a/tree_maker/__init__.py b/tree_maker/__init__.py index e88203331ce136deb4e2e098d848009f547d5567..96553eff7df051bff03362521ab3479eaba70d03 100644 --- a/tree_maker/__init__.py +++ b/tree_maker/__init__.py @@ -11,7 +11,7 @@ from .general import tree_from_yaml from .general import tree_from_json from .general import from_yaml from .general import from_json -from .general import py_to_yaml +from .general import config_to_yaml from .tag import * from .tag_json import * diff --git a/tree_maker/general.py b/tree_maker/general.py index bcdddb0288d0007fd3afe664f6212381a44dca3f..64b2c9ee0a2e28ee0b164d0ee6b692eaa3218e3c 100644 --- a/tree_maker/general.py +++ b/tree_maker/general.py @@ -37,9 +37,9 @@ def from_json(filename, verbose=False): if verbose: print(e) return {} -def py_to_yaml(filename): - '''Convert the *filename*.py (containg dictionaries) to an - equivalent *filename*.yaml''' - import filename as my_dict - with open(filename+'.yaml', 'w') as fid: - yaml.dump(my_dict, fid) +def config_to_yaml(): + '''Convert the config.py (containg dictionaries) to an + equivalent config.yaml''' + import config as my_dict + with open('config.yaml', 'w') as fid: + yaml.dump(my_dict.configuration, fid)