Update pyiceberg/table/__init__.py

Co-authored-by: Fokko Driesprong <[email protected]>
apache · Feb 21, 2025 · 23ed2fc · 23ed2fc
1 parent d8ce99c
commit 23ed2fc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pyiceberg/table/__init__.py b/pyiceberg/table/__init__.py
@@ -1149,6 +1149,11 @@ def upsert(
         Returns:
             An UpsertResult class (contains details of rows updated and inserted)
         """
+        try:
+            import pyarrow as pa
+        except ModuleNotFoundError as e:
+            raise ModuleNotFoundError("For writes PyArrow needs to be installed") from e
+
         from pyiceberg.io.pyarrow import expression_to_pyarrow
         from pyiceberg.table import upsert_util