RuntimeError: Invalid DISPLAY variable

[QiangShiPKU]

Hello,
In the last section "Save SPRING files" of pbmc4k, I have the error:
RuntimeError: Invalid DISPLAY variable

Then it works after I add following code before make_spring_subplot function:
import matplotlib.pyplot as plt
plt.switch_backend('agg')