diff --git a/comfyrecipes/cli.py b/comfyrecipes/cli.py index 5205862..74998e9 100644 --- a/comfyrecipes/cli.py +++ b/comfyrecipes/cli.py @@ -50,7 +50,8 @@ def main() -> None: except KeyboardInterrupt: pass elif args.subcommand == "generate-units": - if not args.force and os.path.isfile(args.directory + "/units.yaml"): + os.chdir(args.directory) + if not args.force and os.path.isfile("units.yaml"): print( "units.yaml already exists, pass --force if you want to overwrite it", file=sys.stderr, @@ -59,7 +60,8 @@ def main() -> None: else: builder.generate_units() elif args.subcommand == "generate-ingredients": - if not args.force and os.path.isfile(args.directory + "/ingredients.yaml"): + os.chdir(args.directory) + if not args.force and os.path.isfile("ingredients.yaml"): print( "ingredients.yaml already exists, pass --force if you want to overwrite it", file=sys.stderr,