Add Optional to remaining types (#37808)

More Optional typing

Signed-off-by: cyy <cyyever@outlook.com>
This commit is contained in:
Yuanyuan Chen
2025-04-28 21:20:45 +08:00
committed by GitHub
parent 1a9188a54e
commit da4ff2a5f5
206 changed files with 553 additions and 531 deletions

View File

@@ -92,12 +92,12 @@ class PatchTSMixerModelTester:
head_dropout: float = 0.2,
# forecast related
prediction_length: int = 16,
out_channels: int = None,
out_channels: Optional[int] = None,
# Classification/regression related
# num_labels: int = 3,
num_targets: int = 3,
output_range: list = None,
head_aggregation: str = None,
output_range: Optional[list] = None,
head_aggregation: Optional[str] = None,
# Trainer related
batch_size=13,
is_training=True,