From 0ba7108aeb44075ca60530af366a2a30f57d52ec Mon Sep 17 00:00:00 2001 From: okiner-3 Date: Wed, 24 Sep 2025 15:32:49 +0900 Subject: [PATCH] remove arg and fix slice of index --- example/example.ipynb | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/example/example.ipynb b/example/example.ipynb index 17910fe..e81d582 100644 --- a/example/example.ipynb +++ b/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",