refactor for conciseness.

tompollard · Jun 7, 2024 · 65733a5 · 65733a5
1 parent 50f3c78
commit 65733a5
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tableone/tableone.py b/tableone/tableone.py
@@ -238,11 +238,7 @@ def __init__(self, data: pd.DataFrame,
 
         # if categorical is set to None, try to automatically detect
         # if empty list is provided, assume there are no categorical variables.
-        if categorical is None:
-            self._categorical = detect_categorical(data[self._columns], groupby)
-        else:
-            self._categorical = categorical
-
+        self._categorical = detect_categorical(data[self._columns], groupby) if categorical is None else categorical
         self._order = order_categorical(data, order)
 
         self._alt_labels = rename