From 62b4fe118f0908cfb4add1c87690d77f3dc57a50 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Tue, 29 Dec 2020 15:40:22 +0100 Subject: [PATCH] prevent overwriting existing config file --- spacy/cli/init_config.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/spacy/cli/init_config.py b/spacy/cli/init_config.py index 90b92787d..55bdacd30 100644 --- a/spacy/cli/init_config.py +++ b/spacy/cli/init_config.py @@ -32,6 +32,7 @@ def init_config_cli( optimize: Optimizations = Opt(Optimizations.efficiency.value, "--optimize", "-o", help="Whether to optimize for efficiency (faster inference, smaller model, lower memory consumption) or higher accuracy (potentially larger and slower model). This will impact the choice of architecture, pretrained weights and related hyperparameters."), gpu: bool = Opt(False, "--gpu", "-G", help="Whether the model can run on GPU. This will impact the choice of architecture, pretrained weights and related hyperparameters."), pretraining: bool = Opt(False, "--pretraining", "-pt", help="Include config for pretraining (with 'spacy pretrain')"), + overwrite: bool = Opt(False, "--overwrite", "-W", help="Whether or not to overwrite the file if it already exists."), # fmt: on ): """ @@ -46,6 +47,9 @@ def init_config_cli( optimize = optimize.value pipeline = string_to_list(pipeline) is_stdout = str(output_file) == "-" + if not is_stdout and output_file.exists() and not overwrite: + msg = Printer(no_print=False) + msg.fail("The provided output file already exists. To overwrite, set the flag -W", exits=1) config = init_config( lang=lang, pipeline=pipeline,