[CHORE] Fix issues from nightly tests (#2926)

notebook checker test failure: https://github.com/Eventual-Inc/Daft/actions/runs/11040289879 property-based test failure: https://github.com/Eventual-Inc/Daft/actions/runs/11040292121
Eventual-Inc · Sep 25, 2024 · 6460a0e · 6460a0e
1 parent b1ea3b9
commit 6460a0e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/daft-dsl/src/resolve_expr.rs b/src/daft-dsl/src/resolve_expr.rs
@@ -388,7 +388,7 @@ pub fn resolve_aggexprs(
 pub fn check_column_name_validity(name: &str, schema: &Schema) -> DaftResult<()> {
     let struct_expr_map = calculate_struct_expr_map(schema);
 
-    let names = if name.contains('*') {
+    let names = if name == "*" || name.ends_with(".*") {
         if let Ok(names) = get_wildcard_matches(name, schema, &struct_expr_map) {
             names
         } else {

diff --git a/tutorials/embeddings/daft_tutorial_embeddings_stackexchange.ipynb b/tutorials/embeddings/daft_tutorial_embeddings_stackexchange.ipynb
@@ -309,12 +309,12 @@
     "    similarity_search(\n",
     "        df[\"embedding\"],\n",
     "        top_embeddings=torch.stack(top_questions[\"embedding\"]),\n",
-    "        top_urls=top_questions[\"URL\"],\n",
+    "        top_urls=top_questions[\"url\"],\n",
     "    ),\n",
     ")\n",
     "\n",
     "df = df.select(\n",
-    "    df[\"URL\"],\n",
+    "    df[\"url\"],\n",
     "    df[\"question_score\"],\n",
     "    df[\"search_result.related_top_question\"].alias(\"related_top_question\"),\n",
     "    df[\"search_result.similarity\"].alias(\"similarity\"),\n",