minor enhancements

Lightning-Universe · Feb 13, 2021 · 8d03c0d · 8d03c0d
1 parent f4e8750
commit 8d03c0d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/flash/tabular/classification/data/data.py b/flash/tabular/classification/data/data.py
@@ -85,11 +85,8 @@ def __init__(
         if categorical_input is None and numerical_input is None:
             raise RuntimeError('Both `categorical_input` and `numerical_input` are None!')
 
-        if categorical_input is None:
-            categorical_input = []
-
-        if numerical_input is None:
-            numerical_input = []
+        categorical_input = categorical_input if categorical_input is not None else []
+        numerical_input = numerical_input if numerical_input is not None else []
 
         if valid_df is not None:
             dfs.append(valid_df)