diff --git a/scripts/checkpoint_to_huggingface.py b/scripts/checkpoint_to_huggingface.py
index d1d7deab..e0ea9b23 100644
--- 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