from pgmpy.models import DiscreteBayesianNetwork
from pgmpy.factors.discrete import TabularCPD
from pgmpy.inference import VariableElimination

model = DiscreteBayesianNetwork([
    ('Smoking', 'LungCancer')
])

cpd_smoking = TabularCPD('Smoking', 2, [[0.5], [0.5]])

cpd_lung = TabularCPD('LungCancer', 2,
                      [[0.9, 0.7],
                       [0.1, 0.3]],
                      evidence=['Smoking'],
                      evidence_card=[2])

model.add_cpds(cpd_smoking, cpd_lung)

infer = VariableElimination(model)

result = infer.query(variables=['LungCancer'], evidence={'Smoking': 1})

print(result)