Fix typing regressions after merging main

Author: tobiasraabe

src/pytask_parallel/config.py

@@ -27,7 +27,8 @@ def pytask_parse_config(config: dict[str, Any]) -> None:
         raise ValueError(msg) from None
 
     if config["n_workers"] == "auto":
-        config["n_workers"] = max(os.cpu_count() - 1, 1)  # type: ignore[operator]
+        cpu_count = os.cpu_count() or 1
+        config["n_workers"] = max(cpu_count - 1, 1)
 
     # If more than one worker is used, and no backend is set, use loky.
     if config["n_workers"] > 1 and config["parallel_backend"] == ParallelBackend.NONE:

src/pytask_parallel/execute.py

@@ -137,7 +137,7 @@ def pytask_execute_build(session: Session) -> bool | None:  # noqa: C901, PLR091
                             newly_collected_reports.append(
                                 ExecutionReport.from_task_and_exception(
                                     task,
-                                    wrapper_result.exc_info,  # type: ignore[arg-type]
+                                    cast("Any", wrapper_result.exc_info),
                                 )
                             )
                             running_tasks.pop(task_name)
@@ -213,7 +213,7 @@ def pytask_execute_task(session: Session, task: PTask) -> Future[WrapperResult]:
         # cloudpickle will pickle it with the function. See cloudpickle#417, pytask#373
         # and pytask#374.
         task_module = get_module(task.function, getattr(task, "path", None))
-        if should_pickle_module_by_value(task_module):
+        if task_module is not None and should_pickle_module_by_value(task_module):
             cloudpickle.register_pickle_by_value(task_module)
 
         return cast("Any", wrapper_func).submit(
@@ -237,7 +237,7 @@ def pytask_execute_task(session: Session, task: PTask) -> Future[WrapperResult]:
         # cloudpickle will pickle it with the function. See cloudpickle#417, pytask#373
         # and pytask#374.
         task_module = get_module(task.function, getattr(task, "path", None))
-        if should_pickle_module_by_value(task_module):
+        if task_module is not None and should_pickle_module_by_value(task_module):
             cloudpickle.register_pickle_by_value(task_module)
 
         return session.config["_parallel_executor"].submit(

src/pytask_parallel/utils.py

@@ -136,7 +136,7 @@ def _parse_future_exception(
     return None if exc is None else (type(exc), exc, exc.__traceback__)
 
 
-def get_module(func: Callable[..., Any], path: Path | None) -> ModuleType:
+def get_module(func: Callable[..., Any], path: Path | None) -> ModuleType | None:
     """Get the module of a python function.
 
     ``functools.partial`` obfuscates the module of the function and
@@ -151,8 +151,8 @@ def get_module(func: Callable[..., Any], path: Path | None) -> ModuleType:
         func = func.func
 
     if path:
-        return inspect.getmodule(func, path.as_posix())  # type: ignore[return-value]
-    return inspect.getmodule(func)  # type: ignore[return-value]
+        return inspect.getmodule(func, path.as_posix())
+    return inspect.getmodule(func)
 
 
 def strip_annotation_locals(task: PTask) -> None:

src/pytask_parallel/wrappers.py

@@ -138,7 +138,11 @@ def wrap_task_in_process(  # noqa: PLR0913
         except Exception:  # noqa: BLE001
             exc_info = sys.exc_info()
             processed_exc_info = _render_traceback_to_string(
-                exc_info,  # type: ignore[arg-type]
+                (
+                    cast("type[BaseException]", exc_info[0]),
+                    cast("BaseException", exc_info[1]),
+                    exc_info[2],
+                ),
                 show_locals,
                 console_options,
             )