nits

Author: ibraheemdev

crates/ty_python_core/src/builder.rs

@@ -3334,9 +3334,10 @@ impl<'ast> Visitor<'ast> for SemanticIndexBuilder<'_, 'ast> {
                 }
             }
             ast::Expr::Lambda(lambda) => {
-                if let Some(current_statement) = self.current_statement_mut() {
-                    current_statement.lambda_exprs.push(lambda);
-                }
+                self.current_statement_mut()
+                    .expect("every lambda expression is part of a statement")
+                    .lambda_exprs
+                    .push(lambda);
 
                 if let Some(parameters) = &lambda.parameters {
                     // The default value of the parameters needs to be evaluated in the

crates/ty_python_semantic/src/types/infer.rs

@@ -48,7 +48,6 @@ use ruff_text_size::Ranged;
 use rustc_hash::{FxHashMap, FxHashSet};
 use salsa;
 use salsa::plumbing::AsId;
-use ty_python_core::statement::StatementInner;
 
 use crate::Db;
 use crate::types::diagnostic::TypeCheckDiagnostics;
@@ -64,6 +63,7 @@ pub(super) use comparisons::UnsupportedComparisonError;
 use ty_python_core::definition::Definition;
 use ty_python_core::expression::Expression;
 use ty_python_core::scope::ScopeId;
+use ty_python_core::statement::StatementInner;
 use ty_python_core::unpack::Unpack;
 use ty_python_core::{ExpressionNodeKey, SemanticIndex, Statement, semantic_index};