diff --git a/pyproject.toml b/pyproject.toml index 744d7149..31a4777b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -64,7 +64,7 @@ version = {attr = "keep_gpu.__version__"} # ----------------------------- # bump-my-version configuration [tool.bumpversion] -current_version = "0.2.0" +current_version = "0.3.0" commit = true tag = true tag_name = "v{new_version}" diff --git a/src/keep_gpu/__init__.py b/src/keep_gpu/__init__.py index 491e9a07..c8a01b37 100644 --- a/src/keep_gpu/__init__.py +++ b/src/keep_gpu/__init__.py @@ -2,4 +2,4 @@ __author__ = """Siyuan Wang""" __email__ = "sywang0227@gmail.com" -__version__ = "0.2.0" +__version__ = "0.3.0" diff --git a/src/keep_gpu/cli.py b/src/keep_gpu/cli.py index d64e1abf..be56c324 100644 --- a/src/keep_gpu/cli.py +++ b/src/keep_gpu/cli.py @@ -42,7 +42,9 @@ def main( try: gpu_id_list = [int(i.strip()) for i in gpu_ids.split(",")] except ValueError: - console.print(f"[bold red]Error: Invalid characters in --gpu-ids '{gpu_ids}'. Please use comma-separated integers.[/bold red]") + console.print( + f"[bold red]Error: Invalid characters in --gpu-ids '{gpu_ids}'. Please use comma-separated integers.[/bold red]" + ) raise typer.Exit(code=1) os.environ["CUDA_VISIBLE_DEVICES"] = ",".join(map(str, gpu_id_list)) logger.info(f"Using specified GPUs: {gpu_id_list}")