remove arg and fix slice of index

Author: okiner-3

example/example.ipynb

@@ -251,7 +251,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "78ade03b-7950-4cd5-9e50-d9f1a949ebac",
    "metadata": {},
    "outputs": [
@@ -278,10 +278,12 @@
    ],
    "source": [
     "pte, lower_bound, upper_bound = estimator.predict_pte(\n",
-    "    target_treatment_arm=1, control_treatment_arm=0, width=1, locations=locations\n",
+    "    target_treatment_arm=1,\n",
+    "    control_treatment_arm=0,\n",
+    "    locations=locations\n",
     ")\n",
     "plot(\n",
-    "    locations,\n",
+    "    locations[:-1],\n",
     "    pte,\n",
     "    lower_bound,\n",
     "    upper_bound,\n",
@@ -515,7 +517,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": null,
    "id": "a1325579-6fca-4f6e-97c8-7b14cab2f984",
    "metadata": {},
    "outputs": [
@@ -544,12 +546,11 @@
     "pte, lower_bound, upper_bound = estimator.predict_pte(\n",
     "    target_treatment_arm=1,\n",
     "    control_treatment_arm=0,\n",
-    "    width=1,\n",
     "    locations=locations,\n",
     "    variance_type=\"simple\",\n",
     ")\n",
     "plot(\n",
-    "    locations,\n",
+    "    locations[:-1],\n",
     "    pte,\n",
     "    lower_bound,\n",
     "    upper_bound,\n",
@@ -560,7 +561,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": null,
    "id": "55a80258-43c3-4fb2-baec-6c2dce3c174a",
    "metadata": {},
    "outputs": [
@@ -589,12 +590,11 @@
     "pte, lower_bound, upper_bound = estimator.predict_pte(\n",
     "    target_treatment_arm=1,\n",
     "    control_treatment_arm=0,\n",
-    "    width=1,\n",
     "    locations=locations,\n",
     "    variance_type=\"moment\",\n",
     ")\n",
     "plot(\n",
-    "    locations,\n",
+    "    locations[:-1],\n",
     "    pte,\n",
     "    lower_bound,\n",
     "    upper_bound,\n",