!pip install whylogs

import whylogs as why
from whylabs_client.api.transactions_api import TransactionsApi
from whylogs.core.schema import DatasetSchema
from whylogs.core.segmentation_partition import segment_on_column
from whylogs.api.writer.whylabs import WhyLabsWriter, WhyLabsTransaction
import os
from uuid import uuid4
from whylogs.datasets import Ecommerce
import numpy as np
import pandas as pd
from datetime import datetime, timedelta, timezone

os.environ["WHYLABS_DEFAULT_ORG_ID"] = "org-XXX"
os.environ["WHYLABS_DEFAULT_DATASET_ID"] = "model-XXX"
os.environ["WHYLABS_API_KEY"] = "XXXX:org-XXX"

dataset = Ecommerce()

daily_batches = dataset.get_inference_data(number_batches=20)
list_daily_batches = list(daily_batches)

columns = ['product','sales_last_week','market_price','rating','category','output_discount','output_prediction','output_score']

df = list_daily_batches[0].data[columns]

df.head()

why.init(force_local=True)

writer = WhyLabsWriter()

transaction_id = writer.start_transaction()
print(f"Started transaction {transaction_id}")
for i in range(5):
  batch_df = list_daily_batches[i].data[columns]
  profile = why.log(batch_df)
  timestamp = datetime.now(tz=timezone.utc) - timedelta(days=i+1)
  profile.set_dataset_timestamp(timestamp)
  status, id = writer.write(profile)
  print(status, id)
writer.commit_transaction()
print("Commiting transaction")

timestamp = datetime.now(tz=timezone.utc) - timedelta(days=2)
timestamp

try:
  with WhyLabsTransaction(writer):
    print("Started transaction")
    for i in range(5):
      batch_df = list_daily_batches[i].data[columns]
      profile = why.log(df)
      profile.set_dataset_timestamp(timestamp)
      status, id = writer.write(profile)
      print(status, id)
except Exception:
  print("Transaction failed")

print("Committed transaction")


schema = DatasetSchema(segments=segment_on_column("output_discount"))
profile = why.log(df, schema=schema)
with WhyLabsTransaction(writer):
  status, id = writer.write(profile)

print(f"{status} {id}")