wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-25 13:30:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix return type

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2019-12-20 11:08:57 +01:00
parent bb701476f2
commit d30a65e8cd
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/dataset/common.py
+5 -3
View File
@@ -1,5 +1,5 @@
from abc import ABC, abstractmethod
from typing import Any, Dict, Iterable, NamedTuple, Sized, List, Optional
from typing import Any, Dict, Iterable, NamedTuple, Sized, List, Optional, Iterator
from pydantic import BaseModel
@@ -38,20 +38,23 @@ class FieldName:
class Dataset(Sized, Iterable[DataEntry], ABC):
@abstractmethod
def __iter__(self) -> Iterable[DataEntry]:
def __iter__(self) -> Iterator[DataEntry]:
pass
@abstractmethod
def __len__(self):
pass
class CategoricalFeatureInfo(BaseModel):
name: str
cardinality: str
class BasicFeatureInfo(BaseModel):
name: str
class MetaData(BaseModel):
freq: str = None
target: Optional[BasicFeatureInfo] = None
@@ -73,4 +76,3 @@ class TrainDatasets(NamedTuple):
metadata: MetaData
train: Dataset
test: Optional[Dataset] = None