modify python source code example #988

Job	Run time
Black format	8s
Lint with flake8 for syntax+other error codes	6s
Lint Notebooks	8s
Type check	3m 24s
Test: Python 3.7 on ubuntu-latest	1m 54s
Test: Python 3.8 on ubuntu-latest	3m 59s
Test: Python 3.9 on ubuntu-latest	3m 16s
Test: Python 3.10 on ubuntu-latest	2m 17s
Test: Python 3.11 on ubuntu-latest	3m 23s
Test: Python 3.7 on macos-latest	1m 38s
Test: Python 3.8 on macos-latest	3m 24s
Test: Python 3.9 on macos-latest	3m 31s
Test: Python 3.10 on macos-latest	4m 38s
Test: Python 3.11 on macos-latest	2m 34s
Test: Python 3.7 on windows-latest	4m 16s
Test: Python 3.8 on windows-latest	7m 9s
Test: Python 3.9 on windows-latest	5m 51s
Test: Python 3.10 on windows-latest	4m 42s
Test: Python 3.11 on windows-latest	5m 44s
	1h 2m 2s

Provide feedback