Minor type fix

Fix type of argument `module` in `load_module` in `dill/session.py`
uqfoundation · Aug 6, 2022 · f196e3c · f196e3c
1 parent 87b8541
commit f196e3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dill/session.py b/dill/session.py
@@ -308,7 +308,7 @@ def _identify_module(file, main=None):
 
 def load_module(
     filename = str(TEMPDIR/'session.pkl'),
-    module: Union[ModuleType, str] = None,
+    module: Optional[Union[ModuleType, str]] = None,
     **kwds
 ) -> Optional[ModuleType]:
     """Update the selected module (default is :py:mod:`__main__`) with