Restrict to trio unless preview is enabled

astral-sh · Jul 8, 2024 · f91175b · f91175b
1 parent 4ef2faf
commit f91175b
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/crates/ruff_linter/src/rules/flake8_async/rules/async_function_with_timeout.rs b/crates/ruff_linter/src/rules/flake8_async/rules/async_function_with_timeout.rs
@@ -6,6 +6,7 @@ use ruff_text_size::Ranged;
 
 use crate::checkers::ast::Checker;
 use crate::rules::flake8_async::helpers::AsyncModule;
+use crate::settings::types::PreviewMode;
 
 /// ## What it does
 /// Checks for `async` functions with a `timeout` argument.
@@ -81,8 +82,17 @@ pub(crate) fn async_function_with_timeout(
         AsyncModule::AsyncIO
     };
 
-    checker.diagnostics.push(Diagnostic::new(
-        AsyncFunctionWithTimeout { module },
-        timeout.range(),
-    ));
+    if matches!(checker.settings.preview, PreviewMode::Disabled) {
+        if matches!(module, AsyncModule::Trio) {
+            checker.diagnostics.push(Diagnostic::new(
+                AsyncFunctionWithTimeout { module },
+                timeout.range(),
+            ));
+        }
+    } else {
+        checker.diagnostics.push(Diagnostic::new(
+            AsyncFunctionWithTimeout { module },
+            timeout.range(),
+        ));
+    }
 }