make lint

Author: kevinjqliu

notebooks/pyiceberg_example.ipynb

@@ -8,11 +8,11 @@
    "outputs": [],
    "source": [
     "# Import required libraries\n",
+    "import pyarrow as pa\n",
+    "\n",
     "import pyiceberg\n",
     "from pyiceberg.catalog import load_catalog\n",
-    "from pyiceberg.table import Table\n",
-    "import pandas as pd\n",
-    "import pyarrow as pa\n",
+    "\n",
     "print(f\"PyIceberg version: {pyiceberg.__version__}\")"
    ]
   },
@@ -34,11 +34,10 @@
    "outputs": [],
    "source": [
     "# Import required libraries\n",
-    "from pyiceberg.catalog import load_catalog\n",
-    "import pyarrow.parquet as pq\n",
-    "import pyarrow.compute as pc\n",
+    "import os\n",
     "import tempfile\n",
-    "import os"
+    "\n",
+    "import pyarrow.compute as pc"
    ]
   },
   {
@@ -63,11 +62,9 @@
     "# Configure and load the catalog\n",
     "catalog = load_catalog(\n",
     "    \"default\",\n",
-    "    **{\n",
-    "        'type': 'sql',\n",
-    "        \"uri\": f\"sqlite:///{warehouse_path}/pyiceberg_catalog.db\",\n",
-    "        \"warehouse\": f\"file://{warehouse_path}\",\n",
-    "    },\n",
+    "    type=\"sql\",\n",
+    "    uri=f\"sqlite:///{warehouse_path}/pyiceberg_catalog.db\",\n",
+    "    warehouse=f\"file://{warehouse_path}\",\n",
     ")\n",
     "\n",
     "print(\"Catalog loaded successfully!\")\n",
@@ -114,15 +111,14 @@
    "outputs": [],
    "source": [
     "# Create sample data using PyArrow\n",
-    "import pyarrow as pa\n",
     "\n",
     "# Sample taxi-like data\n",
     "data = {\n",
-    "    'vendor_id': [1, 2, 1, 2, 1],\n",
-    "    'trip_distance': [1.5, 2.3, 0.8, 5.2, 3.1],\n",
-    "    'fare_amount': [10.0, 15.5, 6.0, 22.0, 18.0],\n",
-    "    'tip_amount': [2.0, 3.0, 1.0, 4.5, 3.5],\n",
-    "    'passenger_count': [1, 2, 1, 3, 2]\n",
+    "    \"vendor_id\": [1, 2, 1, 2, 1],\n",
+    "    \"trip_distance\": [1.5, 2.3, 0.8, 5.2, 3.1],\n",
+    "    \"fare_amount\": [10.0, 15.5, 6.0, 22.0, 18.0],\n",
+    "    \"tip_amount\": [2.0, 3.0, 1.0, 4.5, 3.5],\n",
+    "    \"passenger_count\": [1, 2, 1, 3, 2],\n",
     "}\n",
     "\n",
     "df = pa.table(data)\n",
@@ -313,14 +309,13 @@
    "outputs": [],
    "source": [
     "# List all files in the warehouse\n",
-    "import os\n",
-    "for root, dirs, files in os.walk(warehouse_path):\n",
-    "    level = root.replace(warehouse_path, '').count(os.sep)\n",
-    "    indent = ' ' * 2 * level\n",
-    "    print(f'{indent}{os.path.basename(root)}/')\n",
-    "    subindent = ' ' * 2 * (level + 1)\n",
+    "for root, _dirs, files in os.walk(warehouse_path):\n",
+    "    level = root.replace(warehouse_path, \"\").count(os.sep)\n",
+    "    indent = \" \" * 2 * level\n",
+    "    print(f\"{indent}{os.path.basename(root)}/\")\n",
+    "    subindent = \" \" * 2 * (level + 1)\n",
     "    for file in files:\n",
-    "        print(f'{subindent}{file}')"
+    "        print(f\"{subindent}{file}\")"
    ]
   },
   {