code style

star-whale · Jan 9, 2024 · e5b6ee4 · e5b6ee4
1 parent d5b0b19
commit e5b6ee4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/client/starwhale/api/_impl/argument.py b/client/starwhale/api/_impl/argument.py
@@ -73,8 +73,8 @@ def asdict(self) -> t.Dict[str, t.Any]:
                     r[func][dtype][option.name] = info
         return r
 
-    def asobj(self) -> t.Dict[str, t.Optional[t.Dict[str, t.Dict[str, OptionField]]]]:
-        r: t.Dict[str, t.Optional[t.Dict[str, t.Dict[str, OptionField]]]] = defaultdict(
+    def asobj(self) -> t.Dict[str, t.Dict[str, t.Dict[str, OptionField]]]:
+        r: t.Dict[str, t.Dict[str, t.Dict[str, OptionField]]] = defaultdict(
             lambda: defaultdict(dict)
         )
         for func, dtypes in self._func_related_dataclasses.items():

diff --git a/client/starwhale/api/_impl/job/handler.py b/client/starwhale/api/_impl/job/handler.py
@@ -195,7 +195,7 @@ def get_registered_handlers_with_expanded_needs(
             ctx = ArgumentContext.get_current_context()
             handler_args = ctx.asobj()
             for name, handler in cls._registered_handlers.items():
-                handler.arguments = handler_args.get(handler.name)
+                handler.arguments = handler_args.get(handler.name)  # type: ignore[assignment]
 
                 if name not in expanded_names:
                     expanded_names[name] = set()