I\'m trying to get survival function.
import matplotlib.pyplot as plt
from sksurv.datasets import load_whas500
from s