diff --git a/crates/ruff/src/lib.rs b/crates/ruff/src/lib.rs index e359ffdbd0..e7b13e3bea 100644 --- a/crates/ruff/src/lib.rs +++ b/crates/ruff/src/lib.rs @@ -5,6 +5,7 @@ //! //! [Ruff]: https://github.com/charliermarsh/ruff +pub use codes::{Pyupgrade, RuleCodePrefix}; pub use ruff_python_ast::source_code::round_trip; pub use rule_selector::RuleSelector; pub use rules::pycodestyle::rules::IOError; diff --git a/crates/ruff_benchmark/benches/linter.rs b/crates/ruff_benchmark/benches/linter.rs index b07390edbb..8e0be3a3f8 100644 --- a/crates/ruff_benchmark/benches/linter.rs +++ b/crates/ruff_benchmark/benches/linter.rs @@ -4,7 +4,7 @@ use criterion::{ }; use ruff::linter::lint_only; use ruff::settings::{flags, Settings}; -use ruff::RuleSelector; +use ruff::{Pyupgrade, RuleCodePrefix, RuleSelector}; use ruff_benchmark::{TestCase, TestCaseSpeed, TestFile, TestFileDownloadError}; use std::time::Duration; @@ -73,14 +73,14 @@ fn benchmark_linter(mut group: BenchmarkGroup, settings: &Settings) { group.finish(); } -fn benchmark_default_rules(criterion: &mut Criterion) { - let group = criterion.benchmark_group("linter/default-rules"); - benchmark_linter(group, &Settings::default()); -} - fn benchmark_all_rules(criterion: &mut Criterion) { let settings = Settings { - rules: RuleSelector::All.into_iter().collect(), + rules: RuleSelector::Prefix { + prefix: RuleCodePrefix::Pyupgrade(Pyupgrade::_0), + redirected_from: None, + } + .into_iter() + .collect(), ..Settings::default() }; @@ -88,6 +88,6 @@ fn benchmark_all_rules(criterion: &mut Criterion) { benchmark_linter(group, &settings); } -criterion_group!(default_rules, benchmark_default_rules); +// criterion_group!(default_rules, benchmark_default_rules); criterion_group!(all_rules, benchmark_all_rules); -criterion_main!(default_rules, all_rules); +criterion_main!(all_rules);