Use pyiceberg.__version__ for PuffinWriter created-by default

Switch from importlib.metadata.version('pyiceberg') to the exported
__version__ to match the existing pattern in cli/console.py and avoid
PackageNotFoundError when pyiceberg is not pip-installed.

Co-authored-by: Isaac

Author: moomindani

pyiceberg/table/puffin.py

@@ -14,7 +14,6 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import importlib.metadata
 import io
 import math
 import zlib
@@ -24,6 +23,7 @@
 from pydantic import Field
 from pyroaring import BitMap, FrozenBitMap
 
+from pyiceberg import __version__
 from pyiceberg.io import OutputFile
 from pyiceberg.typedef import IcebergBaseModel
 
@@ -167,7 +167,7 @@ def __init__(self, output_file: OutputFile, created_by: str | None = None) -> No
         self._blobs = []
         self._blob_payloads = []
         self._created_by = (
-            created_by if created_by is not None else f"PyIceberg version {importlib.metadata.version('pyiceberg')}"
+            created_by if created_by is not None else f"PyIceberg version {__version__}"
         )
 
     def set_blob(

tests/table/test_puffin.py

@@ -14,14 +14,14 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import importlib.metadata
 import zlib
 from os import path
 from pathlib import Path
 
 import pytest
 from pyroaring import BitMap
 
+from pyiceberg import __version__
 from pyiceberg.io.pyarrow import PyArrowFileIO
 from pyiceberg.table.puffin import (
     DELETION_VECTOR_MAGIC,
@@ -202,7 +202,7 @@ def test_puffin_writer_default_created_by(tmp_path: Path) -> None:
     writer.finish()
 
     reader = PuffinFile(puffin_path.read_bytes())
-    assert reader.footer.properties["created-by"] == f"PyIceberg version {importlib.metadata.version('pyiceberg')}"
+    assert reader.footer.properties["created-by"] == f"PyIceberg version {__version__}"
 
 
 def test_set_blob_rejects_negative_positions(tmp_path: Path) -> None: