havakv / pycox

Survival analysis with PyTorch
BSD 2-Clause "Simplified" License
803 stars 188 forks source link

ValueError: Need `time` to have same type as `self.durations` #149

Open pooya-mohammadi opened 1 year ago

pooya-mohammadi commented 1 year ago

This error happens when using multi-processing. This error is due to this check that happens in line 55: https://github.com/havakv/pycox/blob/master/pycox/preprocessing/discretization.py#L155

In my humble opinion, I don't see any reason to use is not for validating because a simple != would do the job without checking the identity of the objects.