From f196e3c9cbbba3326064c09760f7129df7f3de4d Mon Sep 17 00:00:00 2001
From: vcwai <victorcwai@gmail.com>
Date: Sun, 7 Aug 2022 05:19:03 +0800
Subject: [PATCH] Minor type fix

Fix type of argument `module` in `load_module` in `dill/session.py`
---
 dill/session.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dill/session.py b/dill/session.py
index 6a037d61..f036c564 100644
--- 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