The author of the Python lifelines
package does not allow for predictions from models when there are time-varying covariates. See this answer for his rationale. I'm not sure if that rationale carries over to the evaluation of proportional hazards, too.
The R survival
package does allow for PH evaluation via Schoenfeld residuals when there are time-varying covariates. As those residuals are only calculated at event times, in a situation with at most 1 event per individual there should be no theoretical problem with calculating those residuals based on the covariate values in place for each individual at risk at each event time. If there might be more than 1 event per individual then you have to decide whether to treat separate events in the same individual separately or to combine them with the collapse
argument to the residuals.coxph()
function.
Even the R survival
package won't provide a formal test of proportional hazards when you have time-varying coefficients in the model. (As I recall, some early versions did claim to perform such tests, but the results were eventually found to be incorrect.) In that situation, the best you can do is to document that the shape of the time-varying coefficient matches the shape of the estimated pattern of residuals over time.