Add type hint for @job and fix for run_locally

src/jobflow/core/job.py

@@ -6,7 +6,7 @@
 import typing
 import warnings
 from dataclasses import dataclass, field
-from typing import cast
+from typing import cast, overload
 
 from monty.json import MSONable, jsanitize
 from typing_extensions import Self
@@ -68,6 +68,16 @@ class JobConfig(MSONable):
     response_manager_config: dict = field(default_factory=dict)
 
 
+@overload
+def job(method: Callable = None) -> Callable[..., Job]:
+    pass
+
+
+@overload
+def job(method: Callable = None, **job_kwargs) -> Callable[..., Callable[..., Job]]:
+    pass
+
+
 def job(method: Callable = None, **job_kwargs):
     """
     Wrap a function to produce a :obj:`Job`.

src/jobflow/managers/local.py

@@ -16,7 +16,7 @@
 def run_locally(
     flow: jobflow.Flow | jobflow.Job | list[jobflow.Job],
     log: bool = True,
-    store: jobflow.JobStore = None,
+    store: jobflow.JobStore | None = None,
     create_folders: bool = False,
     root_dir: str | Path | None = None,
     ensure_success: bool = False,