From 97654a99110f1f55db8eb93fcfa1f6e63f3aa5e4 Mon Sep 17 00:00:00 2001
From: Zanie <contact@zanie.dev>
Date: Wed, 13 Sep 2023 15:48:02 -0500
Subject: [PATCH] Fix invocations

---
 crates/ruff/src/codes.rs  |  1 +
 crates/ruff/src/linter.rs | 25 ++++++++++++-------------
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/crates/ruff/src/codes.rs b/crates/ruff/src/codes.rs
index 1da249d677..31e0c595cd 100644
--- a/crates/ruff/src/codes.rs
+++ b/crates/ruff/src/codes.rs
@@ -868,6 +868,7 @@ pub fn code_to_rule(linter: Linter, code: &str) -> Option<(RuleGroup, Rule)> {
         #[cfg(feature = "test-rules")]
         (Ruff, "900") => (RuleGroup::Unspecified, rules::ruff::rules::StableTestRule),
         #[cfg(feature = "test-rules")]
+        #[allow(deprecated)]
         (Ruff, "901") => (RuleGroup::Nursery, rules::ruff::rules::NurseryTestRule),
         #[cfg(feature = "test-rules")]
         (Ruff, "902") => (RuleGroup::Preview, rules::ruff::rules::PreviewTestRule),
diff --git a/crates/ruff/src/linter.rs b/crates/ruff/src/linter.rs
index 017abc5cee..2f5eaa56d8 100644
--- a/crates/ruff/src/linter.rs
+++ b/crates/ruff/src/linter.rs
@@ -8,16 +8,6 @@ use itertools::Itertools;
 use log::error;
 use rustc_hash::FxHashMap;
 
-use ruff_diagnostics::Diagnostic;
-use ruff_python_ast::imports::ImportMap;
-use ruff_python_ast::PySourceType;
-use ruff_python_codegen::Stylist;
-use ruff_python_index::Indexer;
-use ruff_python_parser::lexer::LexResult;
-use ruff_python_parser::{AsMode, ParseError};
-use ruff_source_file::{Locator, SourceFileBuilder};
-use ruff_text_size::Ranged;
-
 use crate::autofix::{fix_file, FixResult};
 use crate::checkers::ast::check_ast;
 use crate::checkers::filesystem::check_file_path;
@@ -35,6 +25,15 @@ use crate::rules::pycodestyle;
 use crate::settings::{flags, Settings};
 use crate::source_kind::SourceKind;
 use crate::{directives, fs};
+use ruff_diagnostics::Diagnostic;
+use ruff_python_ast::imports::ImportMap;
+use ruff_python_ast::PySourceType;
+use ruff_python_codegen::Stylist;
+use ruff_python_index::Indexer;
+use ruff_python_parser::lexer::LexResult;
+use ruff_python_parser::{AsMode, ParseError};
+use ruff_source_file::{Locator, SourceFileBuilder};
+use ruff_text_size::Ranged;
 
 const CARGO_PKG_NAME: &str = env!("CARGO_PKG_NAME");
 const CARGO_PKG_REPOSITORY: &str = env!("CARGO_PKG_REPOSITORY");
@@ -218,21 +217,21 @@ pub fn check_path(
     {
         if settings.rules.enabled(Rule::StableTestRule) {
             diagnostics.push(Diagnostic::new(
-                Rule::StableTestRule,
+                crate::rules::ruff::rules::StableTestRule,
                 ruff_text_size::TextRange::default(),
             ));
         }
 
         if settings.rules.enabled(Rule::PreviewTestRule) {
             diagnostics.push(Diagnostic::new(
-                Rule::PreviewTestRule,
+                crate::rules::ruff::rules::PreviewTestRule,
                 ruff_text_size::TextRange::default(),
             ));
         }
 
         if settings.rules.enabled(Rule::NurseryTestRule) {
             diagnostics.push(Diagnostic::new(
-                Rule::NurseryTestRule,
+                crate::rules::ruff::rules::NurseryTestRule,
                 ruff_text_size::TextRange::default(),
             ));
         }