wee

pola-rs · Sep 6, 2024 · ed79a06 · ed79a06
1 parent 4f7904b
commit ed79a06
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/py-polars/tests/unit/io/test_delta.py b/py-polars/tests/unit/io/test_delta.py
@@ -474,13 +474,11 @@ def test_write_delta_with_merge(tmp_path: Path) -> None:
         assert merger.predicate == "s.a = t.a"
         assert merger.source_alias == "s"
         assert merger.target_alias == "t"
-    except AttributeError:
+    except AttributeError as err:
         import deltalake
 
-        msg = (
-            f"dl ver {deltalake.__version__},  {TableMerger.__init__.__code__.co_names}"
-        )
-        raise ValueError(msg) from None
+        msg = f"dl ver {deltalake.__version__},  {dir(merger)}"
+        raise ValueError(msg) from err
 
     merger.when_matched_delete(predicate="t.a > 2").execute()