diff --git a/one_click.py b/one_click.py index 0aae427b..3a23246f 100644 --- a/one_click.py +++ b/one_click.py @@ -190,19 +190,18 @@ def update_requirements(initial_installation=False): # Extensions requirements are installed only during the initial install by default. # That can be changed with the INSTALL_EXTENSIONS environment variable. install_extensions = os.environ.get("INSTALL_EXTENSIONS", "false").lower() in ("yes", "y", "true", "1", "t", "on") - if initial_installation or install_extensions: - if not install_extensions: - print_big_message("Will not install extensions due to INSTALL_EXTENSIONS environment variable.") - else: - print("Installing extensions requirements.") - extensions = next(os.walk("extensions"))[1] - for extension in extensions: - if extension in ['superbooga', 'superboogav2']: # No wheels available for requirements - continue + if initial_installation and not install_extensions: + print_big_message("Will not install extensions due to INSTALL_EXTENSIONS environment variable.") + elif initial_installation or install_extensions: + print("Installing extensions requirements.") + extensions = next(os.walk("extensions"))[1] + for extension in extensions: + if extension in ['superbooga', 'superboogav2']: # No wheels available for requirements + continue - extension_req_path = os.path.join("extensions", extension, "requirements.txt") - if os.path.exists(extension_req_path): - run_cmd("python -m pip install -r " + extension_req_path + " --upgrade", assert_success=True, environment=True) + extension_req_path = os.path.join("extensions", extension, "requirements.txt") + if os.path.exists(extension_req_path): + run_cmd("python -m pip install -r " + extension_req_path + " --upgrade", assert_success=True, environment=True) # Detect the PyTorch version torver = torch_version()