test(snowflake): fix snowflake test (#10292)

ibis-project · Oct 9, 2024 · 733e28c · 733e28c
1 parent 54889db
commit 733e28c
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/ibis/backends/snowflake/tests/test_udf.py b/ibis/backends/snowflake/tests/test_udf.py
@@ -126,17 +126,21 @@ def predict_price(
         predicted_price=predict_price(
             (_.carat - _.carat.mean()) / _.carat.std(),
             _.cut.cases(
-                (c, i)
-                for i, c in enumerate(
-                    ("Fair", "Good", "Very Good", "Premium", "Ideal"), start=1
+                *(
+                    (c, i)
+                    for i, c in enumerate(
+                        ("Fair", "Good", "Very Good", "Premium", "Ideal"), start=1
+                    )
                 )
             ),
-            _.color.cases((c, i) for i, c in enumerate("DEFGHIJ", start=1)),
+            _.color.cases(*((c, i) for i, c in enumerate("DEFGHIJ", start=1))),
             _.clarity.cases(
-                (c, i)
-                for i, c in enumerate(
-                    ("I1", "IF", "SI1", "SI2", "VS1", "VS2", "VVS1", "VVS2"),
-                    start=1,
+                *(
+                    (c, i)
+                    for i, c in enumerate(
+                        ("I1", "IF", "SI1", "SI2", "VS1", "VS2", "VVS1", "VVS2"),
+                        start=1,
+                    )
                 )
             ),
         )