Update minor requirements.

arnor-sigurdsson · Jan 3, 2025 · ec17086 · ec17086
1 parent 651e1d3
commit ec17086
Show file tree

Hide file tree

Showing 4 changed files with 243 additions and 329 deletions.
diff --git a/eir/train_utils/criteria.py b/eir/train_utils/criteria.py
@@ -506,7 +506,7 @@ def _cox_ph_loss(
 ) -> torch.Tensor:
     valid_mask = (time != -1) & (event != -1)
 
-    # Early returns if no valid samples or no events
+    # Early returns if no valid samples
     if not valid_mask.any():
         return risk_scores.new_zeros((), requires_grad=True)