Use X | Y in isinstance call instead of (X, Y)

sebp · Jan 1, 2025 · 2e784db · 2e784db
1 parent e0232dd
commit 2e784db
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/sksurv/ensemble/boosting.py b/sksurv/ensemble/boosting.py
@@ -403,7 +403,7 @@ def fit(self, X, y, sample_weight=None):
         Xi = np.column_stack((np.ones(n_samples), X))
 
         self._loss = LOSS_FUNCTIONS[self.loss]()
-        if isinstance(self._loss, (CensoredSquaredLoss, IPCWLeastSquaresError)):
+        if isinstance(self._loss, CensoredSquaredLoss | IPCWLeastSquaresError):
             time = np.log(time)
 
         if not self._is_fitted():
@@ -1262,7 +1262,7 @@ def fit(self, X, y, sample_weight=None, monitor=None):
         # self.loss is guaranteed to be a string
         self._loss = self._get_loss(sample_weight=sample_weight)
 
-        if isinstance(self._loss, (CensoredSquaredLoss, IPCWLeastSquaresError)):
+        if isinstance(self._loss, CensoredSquaredLoss | IPCWLeastSquaresError):
             time = np.log(time)
 
         if self.n_iter_no_change is not None:

diff --git a/sksurv/linear_model/coxph.py b/sksurv/linear_model/coxph.py
@@ -416,7 +416,7 @@ def fit(self, X, y):
         X = validate_data(self, X, ensure_min_samples=2, dtype=np.float64)
         event, time = check_array_survival(X, y)
 
-        if isinstance(self.alpha, (numbers.Real, numbers.Integral)):
+        if isinstance(self.alpha, numbers.Real | numbers.Integral):
             alphas = np.empty(X.shape[1], dtype=float)
             alphas[:] = self.alpha
         else:

diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -90,13 +90,7 @@ def attr_labels(self):
         return ["event", "time"]
 
     def _to_data_frame(self, data, columns):
-        if isinstance(
-            data,
-            (
-                tuple,
-                list,
-            ),
-        ):
+        if isinstance(data, tuple | list):
             data = np.column_stack(data)
         return pd.DataFrame(data, columns=columns)