From a3df6a187e34648f6f059828fd5027eb90ebdf7f Mon Sep 17 00:00:00 2001 From: Ryuta Yoshimatsu Date: Fri, 24 May 2024 22:45:54 +0200 Subject: [PATCH] chronos models integration --- mmf_sa/models/chronosforecast/ChronosPipeline.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mmf_sa/models/chronosforecast/ChronosPipeline.py b/mmf_sa/models/chronosforecast/ChronosPipeline.py index 8123c6e..be0c499 100644 --- a/mmf_sa/models/chronosforecast/ChronosPipeline.py +++ b/mmf_sa/models/chronosforecast/ChronosPipeline.py @@ -1,10 +1,8 @@ from abc import ABC - import pandas as pd import numpy as np import torch from chronos import ChronosPipeline -from transformers import pipeline from sktime.performance_metrics.forecasting import mean_absolute_percentage_error from typing import Iterator from pyspark.sql.functions import collect_list, pandas_udf