WhyLabsWriter refactor

python/examples/advanced/Transaction_Examples.ipynb

@@ -41,7 +41,7 @@
         "from whylabs_client.api.transactions_api import TransactionsApi\n",
         "from whylogs.core.schema import DatasetSchema\n",
         "from whylogs.core.segmentation_partition import segment_on_column\n",
-        "from whylogs.api.writer.whylabs import WhyLabsWriter, WhyLabsTransaction\n",
+        "from whylogs.api.writer.whylabs import WhyLabsWriter\n",
         "import os\n",
         "from uuid import uuid4\n",
         "from whylogs.datasets import Ecommerce\n",
@@ -593,7 +593,7 @@
       "cell_type": "markdown",
       "source": [
         "## Uploading multiple profiles with the same batch timestamp\n",
-        "The `WhyLabsTransaction` context manager can simplify error handling."
+        "The `WhyLabsWriter` can be used as a context manager to simplify transaction error handling and ensure `commit_transaction()` is called."
       ],
       "metadata": {
         "id": "K7hEBIyuzXh0"
@@ -630,7 +630,7 @@
       "cell_type": "code",
       "source": [
         "try:\n",
-        "  with WhyLabsTransaction(writer):\n",
+        "  with WhyLabsWriter() as writer:\n",
         "    print(\"Started transaction\")\n",
         "    for i in range(5):\n",
         "      batch_df = list_daily_batches[i].data[columns]\n",
@@ -670,7 +670,7 @@
     {
       "cell_type": "markdown",
       "source": [
-        "If a `write()` call returns a `False` status, the profile will not be included in the transaction. You might want to retry writing it. If not, that profile will be left out of the transaction, but those successfully written will still be included."
+        "If a `write()` call during the transaction fails (returns a `False` status), the transaction's commit will fail raising an exception."
       ],
       "metadata": {
         "id": "PaAQy-RDftXU"
@@ -692,7 +692,7 @@
       "source": [
         "schema = DatasetSchema(segments=segment_on_column(\"output_discount\"))\n",
         "profile = why.log(df, schema=schema)\n",
-        "with WhyLabsTransaction(writer):\n",
+        "with WhyLabsWriter() as writer:\n",
         "  status, id = writer.write(profile)\n",
         "\n",
         "print(f\"{status} {id}\")"

python/examples/basic/Getting_Started.ipynb

@@ -75,7 +75,7 @@
    "outputs": [],
    "source": [
     "# Note: you may need to restart the kernel to use updated packages.\n",
-    "%pip install whylogs"
+    "%pip install whylogs==1.3.29.dev0"
    ]
   },
   {
@@ -394,7 +394,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "why.write(profile,\"profile.bin\")"
+    "why.write(profile, \"profile.bin\")"
    ]
   },
   {

python/examples/integrations/Feature_Stores_and_whylogs.ipynb

@@ -3183,7 +3183,7 @@
         "    # If new request is from the next day, close logger, save profile in-memory and start logger for the next day\n",
         "    if request_timestamp.day > day_to_log.day:\n",
         "        # let's write our profiles to whylogs_output folder\n",
-        "        why.write(profile,os.path.join(\"whylogs_output\",\"profile_{}_{}_{}.bin\".format(day_to_log.day,day_to_log.month,day_to_log.year)))\n",
+        "        why.write(profile,\"whylogs_output\",\"profile_{}_{}_{}.bin\".format(day_to_log.day,day_to_log.month,day_to_log.year))\n",
         "        day_to_log = request_timestamp.replace(hour=0, minute=0, second=0, microsecond=0)\n",
         "        print(\"Starting logger for day {}....\".format(day_to_log))\n",
         "        profile = None\n",
@@ -3221,7 +3221,7 @@
         "    else:\n",
         "        profile.track(assembled_feature_vector)\n",
         "    \n",
-        "why.write(profile,os.path.join(\"whylogs_output\",\"profile_{}_{}_{}.bin\".format(day_to_log.day,day_to_log.month,day_to_log.year)))"
+        "why.write(profile,\"whylogs_output\",\"profile_{}_{}_{}.bin\".format(day_to_log.day,day_to_log.month,day_to_log.year))"
       ]
     },
     {