lint

ibis-project · Aug 26, 2024 · 054a945 · 054a945
1 parent 1d0281b
commit 054a945
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ibis_ml/utils/_split.py b/ibis_ml/utils/_split.py
@@ -93,8 +93,7 @@ def train_test_split(
         }
     ).mutate(
         **{
-            train_flag: (_[combined_key] + random_str).hash().abs()
-            % num_buckets
+            train_flag: (_[combined_key] + random_str).hash().abs() % num_buckets
             < int((1 - test_size) * num_buckets)
         }
     )