Restrict to trio unless preview is enabled

astral-sh · Jul 8, 2024 · aece7aa · aece7aa
1 parent 4ef2faf
commit aece7aa
Showing 1 changed file with 13 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
@@ -81,8 +81,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(),
+        ));
+    }
 }