Update type hints

openclimatefix · Aug 6, 2024 · e2b27ac · e2b27ac
1 parent 1a5cfdc
commit e2b27ac
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/scripts/checkpoint_to_huggingface.py b/scripts/checkpoint_to_huggingface.py
@@ -7,7 +7,6 @@
 """
 
 import tempfile
-from typing import Optional
 
 import typer
 import wandb
@@ -21,8 +20,8 @@
 def push_to_huggingface(
     checkpoint_dir_paths: list[str],
     val_best: bool = True,
-    wandb_ids: list[str] = [],
-    local_path: Optional[str] = None,
+    wandb_ids: list[str] | None = [],
+    local_path: str | None = None,
     push_to_hub: bool = True,
 ):
     """Push a local model to pvnet_v2 huggingface model repo