make formatting mandatory & reformat kaplan meier test

Novartis · Sep 26, 2024 · 71ffc30 · 71ffc30
1 parent 07d085d
commit 71ffc30
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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)