diff --git a/tests/test_torchfix.py b/tests/test_torchfix.py index 46f1e71..a31bc1b 100644 --- a/tests/test_torchfix.py +++ b/tests/test_torchfix.py @@ -120,6 +120,3 @@ def test_no_python_files(tmp_path): ) # Check that the script exits successfully assert result.returncode == 0 - - # Check that the correct message is printed - assert "No Python files with torch imports found." in result.stderr diff --git a/torchfix/__main__.py b/torchfix/__main__.py index 781b8d1..4964e01 100644 --- a/torchfix/__main__.py +++ b/torchfix/__main__.py @@ -99,7 +99,6 @@ def main() -> None: break if not torch_files: - print("No Python files with torch imports found.", file=sys.stderr) return config = TorchCodemodConfig() config.select = list(process_error_code_str(args.select))