Skip to content

Commit 87946cf

Browse files
committed
Fix literal
1 parent 8bd88bb commit 87946cf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/lightly_train/_commands/export_task.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def export_onnx(
7575
num_channels: int = 3,
7676
height: int = 224,
7777
width: int = 224,
78-
precision: str = Literal["32-true", "16-true"],
78+
precision: Literal["32-true", "16-true"] = "32-true",
7979
verify: bool = True,
8080
overwrite: bool = False,
8181
format_args: dict[str, Any] | None = None,
@@ -92,7 +92,7 @@ def _export_task(
9292
num_channels: int = 3,
9393
height: int = 224,
9494
width: int = 224,
95-
precision: str = Literal["32-true", "16-true"],
95+
precision: Literal["32-true", "16-true"] = "32-true",
9696
verify: bool = True,
9797
overwrite: bool = False,
9898
format_args: dict[str, Any] | None = None,

0 commit comments

Comments
 (0)