-ax.fill_between(u.numpy(), (mean-std).detach().numpy(), (mean+std).detach().numpy(), color = '#e0e0e0')
-ax.plot(u.numpy(), mean.detach().numpy(), color = 'red')
+ax.fill_between(
+ u.numpy(),
+ (mean - std).detach().numpy(),
+ (mean + std).detach().numpy(),
+ color="#e0e0e0",
+)
+ax.plot(u.numpy(), mean.detach().numpy(), color="red")