if tau is not None:
survival_test = survival_test[test_time < tau]
_, _, estimate = _check_inputs(
test_event, test_time, estimate)
cens = CensoringDistributionEstimator()
After Change
if tau is not None:
survival_test = survival_test[test_time < tau]
estimate = check_array(estimate, ensure_2d=False)
check_consistent_length(test_event, test_time, estimate)
cens = CensoringDistributionEstimator()
cens.fit(survival_train)