From 25a7e450ed76c7a0820834a3a91134476b1b8253 Mon Sep 17 00:00:00 2001
From: vcwai <victorcwai@gmail.com>
Date: Sun, 14 Aug 2022 03:19:08 +0800
Subject: [PATCH] Minor type fix (#535)

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 aaf36ec9..6acdd432 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