Skip to content

Commit

Permalink
Merge pull request #35 from databricks-industry-solutions/integrate-c…
Browse files Browse the repository at this point in the history
…hronos

chronos models integration
  • Loading branch information
ryuta-yoshimatsu authored May 24, 2024
2 parents 3918c6c + a3df6a1 commit fef34ce
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions mmf_sa/models/chronosforecast/ChronosPipeline.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down

0 comments on commit fef34ce

Please sign in to comment.