Spaces:
Running
Running
Update registry.py
Browse files- registry.py +13 -2
registry.py
CHANGED
@@ -6,7 +6,15 @@ class FilterRegistry:
|
|
6 |
self.filters = {}
|
7 |
self.params_map = {}
|
8 |
|
9 |
-
def register(self, name,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
def decorator(func):
|
11 |
self.filters[name] = func
|
12 |
sig = inspect.signature(func)
|
@@ -16,7 +24,10 @@ class FilterRegistry:
|
|
16 |
continue
|
17 |
params[param.name] = {
|
18 |
'type': param.annotation,
|
19 |
-
'default': param.default if param.default != inspect.Parameter.empty else defaults.get(param.name)
|
|
|
|
|
|
|
20 |
}
|
21 |
self.params_map[name] = params
|
22 |
return func
|
|
|
6 |
self.filters = {}
|
7 |
self.params_map = {}
|
8 |
|
9 |
+
def register(self, name, defaults=None, min_vals=None, max_vals=None, step_vals=None):
|
10 |
+
if defaults is None:
|
11 |
+
defaults = {}
|
12 |
+
if min_vals is None:
|
13 |
+
min_vals = {}
|
14 |
+
if max_vals is None:
|
15 |
+
max_vals = {}
|
16 |
+
if step_vals is None:
|
17 |
+
step_vals = {}
|
18 |
def decorator(func):
|
19 |
self.filters[name] = func
|
20 |
sig = inspect.signature(func)
|
|
|
24 |
continue
|
25 |
params[param.name] = {
|
26 |
'type': param.annotation,
|
27 |
+
'default': param.default if param.default != inspect.Parameter.empty else defaults.get(param.name),
|
28 |
+
'min': min_vals.get(param.name),
|
29 |
+
'max': max_vals.get(param.name),
|
30 |
+
'step': step_vals.get(param.name)
|
31 |
}
|
32 |
self.params_map[name] = params
|
33 |
return func
|