allow await outside async function def for notebooks

Reviewed By: ndmitchell

Differential Revision: D85445448

fbshipit-source-id: b44b2de1c1e7c8b7ab4a45c25de8ecff106c0475

Author: yangdanny97

crates/pyrefly_python/src/module_path.rs

@@ -225,6 +225,10 @@ impl ModulePath {
         self.style() == ModuleStyle::Interface
     }
 
+    pub fn is_notebook(&self) -> bool {
+        self.as_path().extension() == Some("ipynb".as_ref())
+    }
+
     /// Attempt to match the given [`ModuleName`]'s components to this `ModulePath`,
     /// returning the directory that is the import root for the `ModuleName`, if
     /// *all* module components could be matched to directories. `ModulePath`s

pyrefly/lib/binding/expr.rs

@@ -723,7 +723,7 @@ impl<'a> BindingsBuilder<'a> {
             }
             Expr::Await(x) => {
                 self.ensure_expr(&mut x.value, usage);
-                if !self.scopes.is_in_async_def() {
+                if !self.scopes.is_in_async_def() && !self.module_info.path().is_notebook() {
                     self.error(
                         x.range(),
                         ErrorInfo::Kind(ErrorKind::InvalidSyntax),