@@ -27,30 +27,20 @@ def geometry_type_filter(geom_type: GeometryType | str) -> list:
2727 return ["==" , get_geometry_type (), GeometryType (geom_type ).value ]
2828
2929
30- def interpolate (
31- property : str | list , stops : list , outputs : list , type : list = ["linear" ]
32- ) -> list :
30+ def interpolate (property : str | list , stops : list , outputs : list , type : list = ["linear" ]) -> list :
3331 assert len (stops ) == len (outputs )
3432 return [
3533 "interpolate" ,
3634 type ,
3735 get_column (property ) if isinstance (property , str ) else property ,
38- ] + list (
39- itertools .chain .from_iterable (
40- [[stop , output ] for stop , output in zip (stops , outputs )]
41- )
42- )
36+ ] + list (itertools .chain .from_iterable ([[stop , output ] for stop , output in zip (stops , outputs )]))
4337
4438
4539def match_expr (column : str , categories : list , outputs : list [T ], fallback : T ) -> list :
4640 assert len (categories ) == len (outputs )
4741 return (
4842 ["match" , get_column (column )]
49- + list (
50- itertools .chain .from_iterable (
51- [[category , output ] for category , output in zip (categories , outputs )]
52- )
53- )
43+ + list (itertools .chain .from_iterable ([[category , output ] for category , output in zip (categories , outputs )]))
5444 + [fallback ]
5545 )
5646
@@ -70,18 +60,12 @@ def step_expr(
7060 "step" ,
7161 get_column (property ) if isinstance (property , str ) else property ,
7262 ]
73- + list (
74- itertools .chain .from_iterable (
75- [[output , stop ] for output , stop in zip (outputs , stops )]
76- )
77- )
63+ + list (itertools .chain .from_iterable ([[output , stop ] for output , stop in zip (outputs , stops )]))
7864 + [fallback ]
7965 )
8066
8167
82- def quantile_step_expr (
83- column : str , probs : list , outputs : list [T ], fallback : T , values : list
84- ) -> list :
68+ def quantile_step_expr (column : str , probs : list , outputs : list [T ], fallback : T , values : list ) -> list :
8569 assert len (probs ) == len (outputs )
8670 try :
8771 import numpy as np
@@ -98,14 +82,10 @@ def quantile_step_expr(
9882# -----
9983
10084
101- def color_quantile_step_expr (
102- column : str , probs : list , values : Any , cmap = "viridis"
103- ) -> list :
85+ def color_quantile_step_expr (column : str , probs : list , values : Any , cmap = "viridis" ) -> list :
10486 n = len (probs )
10587 colors = color_brewer (cmap , n + 1 )
106- return quantile_step_expr (
107- column , probs , outputs = colors [0 :n ], fallback = colors [- 1 ], values = values
108- )
88+ return quantile_step_expr (column , probs , outputs = colors [0 :n ], fallback = colors [- 1 ], values = values )
10989
11090
11191def color_step_expr (column : str , stops : list , cmap = "viridis" ) -> list :
@@ -143,8 +123,6 @@ def filter_expr(column: str, operator: str, value: Any) -> list:
143123 return [operator , get_column (column ), value ]
144124
145125
146- def range_filter (
147- column , values : tuple | list , operators : tuple | list = (">=" , "<=" )
148- ) -> list :
126+ def range_filter (column , values : tuple | list , operators : tuple | list = (">=" , "<=" )) -> list :
149127 assert len (values ) == len (operators ) == 2
150128 return ["all" ] + [[operators [i ], get_column (column ), values [i ]] for i in range (2 )]
0 commit comments