diff --git a/ts2vec.py b/ts2vec.py index de909bd5..e1e2338a 100644 --- a/ts2vec.py +++ b/ts2vec.py @@ -4,7 +4,7 @@ import numpy as np from models import TSEncoder from models.losses import hierarchical_contrastive_loss -from utils import take_per_row, split_with_nan, centerize_vary_length_series, torch_pad_nan +from .utils import take_per_row, split_with_nan, centerize_vary_length_series, torch_pad_nan import math class TS2Vec: