Skip to content

Commit

Permalink
Update _describer.py
Browse files Browse the repository at this point in the history
Signed-off-by: Zishen Wang <[email protected]>
  • Loading branch information
zz11ss11zz authored Sep 21, 2024
1 parent 228f256 commit d9b9e1e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/maml/base/_describer.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@ def transform(self, objs: list[Any]) -> Any:
multi_output = self._is_multi_output()
if not multi_output:
features = [features]
batched_features = [self.feature_batch(i) for i in list(*zip(features, strict=False))] # type: ignore
batched_features = [self.feature_batch(i) for i in list(*zip(features))] # type: ignore
return batched_features if multi_output else batched_features[0]

def _is_multi_output(self) -> bool:
Expand Down

0 comments on commit d9b9e1e

Please sign in to comment.