Fix Log Entry to Config File Bug

casanovo/casanovo.py

@@ -41,10 +41,9 @@
 import tqdm
 from lightning.pytorch import seed_everything
 
-from . import __version__
-from . import utils
-from .denovo import ModelRunner
+from . import __version__, utils
 from .config import Config
+from .denovo import ModelRunner
 
 logger = logging.getLogger("casanovo")
 click.rich_click.USE_MARKDOWN = True
@@ -337,21 +336,46 @@ def version() -> None:
 
 
 @main.command()
+@click.option(
+    "-d",
+    "--output_dir",
+    help="The destination directory for log and config file.",
+    type=click.Path(dir_okay=True),
+    required=False,
+)
 @click.option(
     "-o",
-    "--output",
-    help="The output configuration file.",
-    default="casanovo.yaml",
+    "--output_root",
+    help="The root name for log and config file.",
     type=click.Path(dir_okay=False),
+    required=False,
 )
-def configure(output: Path) -> None:
+@click.option(
+    "-f",
+    "--force_overwrite",
+    help="Whether to overwrite output files.",
+    is_flag=True,
+    show_default=True,
+    default=False,
+)
+def configure(
+    output_dir: str, output_root: str, force_overwrite: bool
+) -> None:
     """Generate a Casanovo configuration file to customize.
 
     The casanovo configuration file is in the YAML format.
     """
-    Config.copy_default(str(output))
-    setup_logging(output, "info")
-    logger.info(f"Wrote {output}\n")
+    output_path, _ = _setup_output(
+        output_dir, output_root, force_overwrite, "info"
+    )
+    config_fname = output_root if output_root is not None else "casanovo"
+    config_fname = Path(config_fname).with_suffix(".yaml")
+    if not force_overwrite:
+        utils.check_dir_file_exists(output_path, str(config_fname))
+
+    config_path = str(output_path / config_fname)
+    Config.copy_default(config_path)
+    logger.info(f"Wrote {config_path}\n")
 
 
 def setup_logging(