model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "matplotlib",8#     "protobuf",9#     "torch",10#     "sentencepiece",11#     "torchvision",12#     "transformers",13#     "timm",14#     "diffusers",15#     "sentence-transformers",16#     "accelerate",17#     "peft",18#     "slack-sdk",19# ]20# ///21 22try:23    import pandas as pd24    from chronos import BaseChronosPipeline25    26    pipeline = BaseChronosPipeline.from_pretrained("amazon/chronos-2", device_map="cuda")27    28    # Load historical data29    context_df = pd.read_csv("https://autogluon.s3.us-west-2.amazonaws.com/datasets/timeseries/misc/AirPassengers.csv")30    31    # Generate predictions32    pred_df = pipeline.predict_df(33        context_df,34        prediction_length=36,  # Number of steps to forecast35        quantile_levels=[0.1, 0.5, 0.9],  # Quantiles for probabilistic forecast36        id_column="item_id",  # Column identifying different time series37        timestamp_column="Month",  # Column with datetime information38        target="#Passengers",  # Column(s) with time series values to predict39    )40    with open('amazon_chronos-2_0.txt', 'w', encoding='utf-8') as f:41        f.write('Everything was good in amazon_chronos-2_0.txt')42except Exception as e:43    import os44    from slack_sdk import WebClient45    client = WebClient(token=os.environ['SLACK_TOKEN'])46    client.chat_postMessage(47        channel='#hub-model-metadata-snippets-sprint',48        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/amazon_chronos-2_0.txt|amazon_chronos-2_0.txt>',49    )50 51    with open('amazon_chronos-2_0.txt', 'a', encoding='utf-8') as f:52        import traceback53        f.write('''```CODE: 54import pandas as pd55from chronos import BaseChronosPipeline56 57pipeline = BaseChronosPipeline.from_pretrained("amazon/chronos-2", device_map="cuda")58 59# Load historical data60context_df = pd.read_csv("https://autogluon.s3.us-west-2.amazonaws.com/datasets/timeseries/misc/AirPassengers.csv")61 62# Generate predictions63pred_df = pipeline.predict_df(64    context_df,65    prediction_length=36,  # Number of steps to forecast66    quantile_levels=[0.1, 0.5, 0.9],  # Quantiles for probabilistic forecast67    id_column="item_id",  # Column identifying different time series68    timestamp_column="Month",  # Column with datetime information69    target="#Passengers",  # Column(s) with time series values to predict70)71```72 73ERROR: 74''')75        traceback.print_exc(file=f)76    77finally:78    from huggingface_hub import upload_file79    upload_file(80        path_or_fileobj='amazon_chronos-2_0.txt',81        repo_id='model-metadata/code_execution_files',82        path_in_repo='amazon_chronos-2_0.txt',83        repo_type='dataset',84    )85