Fix CI, fix existing table test

Author: datapythonista

pandas/io/iceberg.py

@@ -8,12 +8,12 @@
 from pandas import DataFrame
 
 if TYPE_CHECKING:
-    from pyiceberg.catalog import Catalog  # noqa: TC004
+    from pyiceberg.catalog import Catalog
 
 
 def _get_catalog(
     catalog_name: str | None, catalog_properties: dict[str, Any] | None
-) -> Catalog:
+) -> "Catalog":
     pyiceberg_catalog = import_optional_dependency("pyiceberg.catalog")
     if catalog_properties is None:
         catalog_properties = {}

pandas/tests/io/test_iceberg.py

@@ -185,10 +185,11 @@ def test_write_existing_table(self, catalog):
         )
         new = pd.DataFrame(
             {
-                "A": [4, 5],
-                "B": ["bar", "foobar"],
+                "A": [1, 2, 3],
+                "B": ["foo", "foo", "foo"],
             }
         )
+        expected = pd.concat([original, new], ignore_index=True)
         new.to_iceberg(
             "ns.my_table",
             catalog_properties={"uri": catalog.uri},
@@ -198,4 +199,4 @@ def test_write_existing_table(self, catalog):
             "ns.my_table",
             catalog_properties={"uri": catalog.uri},
         )
-        tm.assert_frame_equal(result, pd.concat([original, new]))
+        tm.assert_frame_equal(result, expected)