We can install contagion with pip:
contagion
pip install contagion
Once contagion is installed, import NetworkX and contagion:
NetworkX
import networkx from contagion import contagion
a Python package supporting node immunization and network contagion simulation