mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-09-11 12:42:03 +08:00
initial gluonts dependency
This commit is contained in:
@@ -16,7 +16,7 @@ from abc import ABC, abstractmethod
|
||||
from functools import reduce
|
||||
from typing import Callable, Iterator, Iterable, List
|
||||
|
||||
from pts.core.component import validated
|
||||
from gluonts.core.component import validated
|
||||
from pts.dataset import DataEntry
|
||||
|
||||
MAX_IDLE_TRANSFORMS = 100
|
||||
@@ -43,6 +43,7 @@ class Chain(Transformation):
|
||||
"""
|
||||
Chain multiple transformations together.
|
||||
"""
|
||||
|
||||
@validated()
|
||||
def __init__(self, trans: List[Transformation]) -> None:
|
||||
self.transformations = []
|
||||
|
||||
Reference in New Issue
Block a user