Auto-format

This commit is contained in:
Ines Montani 2020-10-02 01:12:30 +02:00
parent 6b94cee468
commit e59ecb12c0
1 changed files with 3 additions and 1 deletions

View File

@ -9,7 +9,9 @@ def FeatureExtractor(columns: List[Union[int, str]]) -> Model[List[Doc], List[In
return Model("extract_features", forward, attrs={"columns": columns})
def forward(model: Model[List[Doc], List[Ints2d]], docs, is_train: bool) -> Tuple[List[Ints2d], Callable]:
def forward(
model: Model[List[Doc], List[Ints2d]], docs, is_train: bool
) -> Tuple[List[Ints2d], Callable]:
columns = model.attrs["columns"]
features: List[Ints2d] = []
for doc in docs: