Type inference python jupyter notebook