Skip to content

Commit 84b8c55

Browse files
author
Beat Buesser
committed
Update default value for nb_classes
Signed-off-by: Beat Buesser <[email protected]>
1 parent b970b30 commit 84b8c55

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

art/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -670,7 +670,7 @@ def floats_to_one_hot(labels: np.ndarray):
670670

671671

672672
def check_and_transform_label_format(
673-
labels: np.ndarray, nb_classes: Optional[int] = None, return_one_hot: bool = True
673+
labels: np.ndarray, nb_classes: Optional[int], return_one_hot: bool = True
674674
) -> np.ndarray:
675675
"""
676676
Check label format and transform to one-hot-encoded labels if necessary

0 commit comments

Comments
 (0)