diff --git a/.github/workflows/codeqc.yml b/.github/workflows/codeqc.yml index 889c8c3..b9cdf09 100644 --- a/.github/workflows/codeqc.yml +++ b/.github/workflows/codeqc.yml @@ -46,7 +46,7 @@ jobs: shell: bash -l {0} run: | conda activate torchsurv - ./dev/codeqc.sh + ./dev/codeqc.sh check - name: Tests shell: bash -l {0} diff --git a/tests/test_kaplan_meier.py b/tests/test_kaplan_meier.py index cfff910..62e0a5d 100644 --- a/tests/test_kaplan_meier.py +++ b/tests/test_kaplan_meier.py @@ -208,9 +208,9 @@ def test_kaplan_meier_prediction_error_raised(self): for batch in batch_container.batches: (train_time, train_event, test_time, *_) = batch - train_event[ - -1 - ] = False # if last event is censoring, the last KM is > 0 and it cannot predict beyond this time + train_event[-1] = ( + False # if last event is censoring, the last KM is > 0 and it cannot predict beyond this time + ) km = KaplanMeierEstimator() km(train_event, train_time, censoring_dist=False)