@@ -160,78 +160,3 @@ def test_amd_linux_rocm_version_controls_url(self, tmp_path, rocm_version, expec
160160 cmd = _get_torch_install_cmd (mock_run .call_args_list )
161161 assert expected_url in cmd
162162
163- def test_nvidia_linux_no_extra_index (self , tmp_path ):
164- repo_dir = str (tmp_path )
165- (tmp_path / "requirements.txt" ).write_text ("some-package\n " )
166-
167- with patch ("comfy_cli.command.install.subprocess.run" , return_value = MagicMock (returncode = 0 )) as mock_run :
168- install .pip_install_comfyui_dependencies (
169- repo_dir ,
170- gpu = GPU_OPTION .NVIDIA ,
171- plat = constants .OS .LINUX ,
172- cuda_version = constants .CUDAVersion .v12_6 ,
173- skip_torch_or_directml = False ,
174- skip_requirement = False ,
175- python = "/usr/bin/python" ,
176- )
177-
178- cmd = _get_torch_install_cmd (mock_run .call_args_list )
179- assert "--extra-index-url" not in cmd
180- assert "--index-url" not in cmd
181-
182- def test_nvidia_windows_uses_extra_index_url (self , tmp_path ):
183- repo_dir = str (tmp_path )
184- (tmp_path / "requirements.txt" ).write_text ("some-package\n " )
185-
186- with patch ("comfy_cli.command.install.subprocess.run" , return_value = MagicMock (returncode = 0 )) as mock_run :
187- install .pip_install_comfyui_dependencies (
188- repo_dir ,
189- gpu = GPU_OPTION .NVIDIA ,
190- plat = constants .OS .WINDOWS ,
191- cuda_version = constants .CUDAVersion .v12_6 ,
192- skip_torch_or_directml = False ,
193- skip_requirement = False ,
194- python = "/usr/bin/python" ,
195- )
196-
197- cmd = _get_torch_install_cmd (mock_run .call_args_list )
198- assert "--extra-index-url" in cmd
199- assert "https://download.pytorch.org/whl/cu126" in cmd
200-
201- def test_cpu_uses_extra_index_url (self , tmp_path ):
202- repo_dir = str (tmp_path )
203- (tmp_path / "requirements.txt" ).write_text ("some-package\n " )
204-
205- with patch ("comfy_cli.command.install.subprocess.run" , return_value = MagicMock (returncode = 0 )) as mock_run :
206- install .pip_install_comfyui_dependencies (
207- repo_dir ,
208- gpu = None ,
209- plat = constants .OS .LINUX ,
210- cuda_version = constants .CUDAVersion .v12_6 ,
211- skip_torch_or_directml = False ,
212- skip_requirement = False ,
213- python = "/usr/bin/python" ,
214- )
215-
216- cmd = _get_torch_install_cmd (mock_run .call_args_list )
217- assert "--extra-index-url" in cmd
218- assert "https://download.pytorch.org/whl/cpu" in cmd
219-
220- def test_amd_windows_installs_directml_not_rocm (self , tmp_path ):
221- repo_dir = str (tmp_path )
222- (tmp_path / "requirements.txt" ).write_text ("some-package\n " )
223-
224- with patch ("comfy_cli.command.install.subprocess.run" , return_value = MagicMock (returncode = 0 )) as mock_run :
225- install .pip_install_comfyui_dependencies (
226- repo_dir ,
227- gpu = GPU_OPTION .AMD ,
228- plat = constants .OS .WINDOWS ,
229- cuda_version = constants .CUDAVersion .v12_6 ,
230- skip_torch_or_directml = False ,
231- skip_requirement = False ,
232- python = "/usr/bin/python" ,
233- )
234-
235- cmds = [c [0 ][0 ] for c in mock_run .call_args_list ]
236- assert any ("torch-directml" in c for c in cmds )
237- assert not any ("--index-url" in c for c in cmds )
0 commit comments