Fix code completion

Author: FBumann

docs/examples/advanced.ipynb

@@ -17,9 +17,8 @@
    "source": [
     "import numpy as np\n",
     "import pandas as pd\n",
-    "import xarray as xr\n",
     "\n",
-    "import xarray_plotly"
+    "from xarray_plotly import DataArray  # IDE completion for .plotly"
    ]
   },
   {
@@ -37,7 +36,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "da = xr.DataArray(\n",
+    "da = DataArray(\n",
     "    np.random.randn(30, 3).cumsum(axis=0) + 15,\n",
     "    dims=[\"time\", \"station\"],\n",
     "    coords={\n",
@@ -377,7 +376,7 @@
     "fig = monthly_mean.plotly.line(\n",
     "    title=\"Monthly Climatology\",\n",
     ")\n",
-    "fig.update_xaxes(tickmode=\"array\", tickvals=list(range(1, 13)), \n",
+    "fig.update_xaxes(tickmode=\"array\", tickvals=list(range(1, 13)),\n",
     "                 ticktext=[\"Jan\", \"Feb\", \"Mar\", \"Apr\", \"May\", \"Jun\",\n",
     "                          \"Jul\", \"Aug\", \"Sep\", \"Oct\", \"Nov\", \"Dec\"])\n",
     "fig"

docs/examples/plot-types.ipynb

@@ -17,9 +17,8 @@
    "source": [
     "import numpy as np\n",
     "import pandas as pd\n",
-    "import xarray as xr\n",
     "\n",
-    "import xarray_plotly"
+    "from xarray_plotly import DataArray  # IDE completion for .plotly"
    ]
   },
   {
@@ -40,7 +39,7 @@
     "np.random.seed(42)\n",
     "\n",
     "# Time series data\n",
-    "da_ts = xr.DataArray(\n",
+    "da_ts = DataArray(\n",
     "    np.random.randn(30, 3).cumsum(axis=0),\n",
     "    dims=[\"time\", \"category\"],\n",
     "    coords={\n",
@@ -51,7 +50,7 @@
     ")\n",
     "\n",
     "# 2D grid data\n",
-    "da_2d = xr.DataArray(\n",
+    "da_2d = DataArray(\n",
     "    np.random.rand(20, 30),\n",
     "    dims=[\"lat\", \"lon\"],\n",
     "    coords={\n",
@@ -62,7 +61,7 @@
     ")\n",
     "\n",
     "# Categorical data\n",
-    "da_cat = xr.DataArray(\n",
+    "da_cat = DataArray(\n",
     "    np.random.rand(4, 3) * 100,\n",
     "    dims=[\"product\", \"region\"],\n",
     "    coords={\n",
@@ -161,7 +160,7 @@
    "outputs": [],
    "source": [
     "# Use absolute values for stacking to make sense\n",
-    "da_positive = xr.DataArray(\n",
+    "da_positive = DataArray(\n",
     "    np.abs(np.random.randn(30, 3)) * 10,\n",
     "    dims=[\"time\", \"source\"],\n",
     "    coords={\n",
@@ -229,7 +228,7 @@
    "outputs": [],
    "source": [
     "# Create data with more samples\n",
-    "da_dist = xr.DataArray(\n",
+    "da_dist = DataArray(\n",
     "    np.random.randn(100, 4) + np.array([0, 1, 2, 3]),\n",
     "    dims=[\"sample\", \"group\"],\n",
     "    coords={\"group\": [\"Control\", \"Treatment A\", \"Treatment B\", \"Treatment C\"]},\n",
@@ -295,7 +294,7 @@
    "outputs": [],
    "source": [
     "# 3D data for faceting\n",
-    "da_3d = xr.DataArray(\n",
+    "da_3d = DataArray(\n",
     "    np.random.randn(30, 3, 2).cumsum(axis=0),\n",
     "    dims=[\"time\", \"city\", \"scenario\"],\n",
     "    coords={\n",
@@ -330,11 +329,11 @@
    "outputs": [],
    "source": [
     "# Create monthly data\n",
-    "da_monthly = xr.DataArray(\n",
+    "da_monthly = DataArray(\n",
     "    np.random.rand(12, 4) * 100,\n",
     "    dims=[\"month\", \"product\"],\n",
     "    coords={\n",
-    "        \"month\": [\"Jan\", \"Feb\", \"Mar\", \"Apr\", \"May\", \"Jun\", \n",
+    "        \"month\": [\"Jan\", \"Feb\", \"Mar\", \"Apr\", \"May\", \"Jun\",\n",
     "                  \"Jul\", \"Aug\", \"Sep\", \"Oct\", \"Nov\", \"Dec\"],\n",
     "        \"product\": [\"A\", \"B\", \"C\", \"D\"],\n",
     "    },\n",