Upload renderers.py with huggingface_hub
Browse files- renderers.py +13 -2
renderers.py
CHANGED
@@ -19,8 +19,19 @@ class Renderer(ABC):
|
|
19 |
class RenderTemplate(Renderer, StreamInstanceOperator):
|
20 |
template: Template
|
21 |
random_reference: bool = False
|
|
|
22 |
|
23 |
def process(self, instance: Dict[str, Any], stream_name: str = None) -> Dict[str, Any]:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
24 |
inputs = instance.pop("inputs")
|
25 |
outputs = instance.pop("outputs")
|
26 |
|
@@ -58,8 +69,8 @@ class RenderDemonstrations(RenderTemplate):
|
|
58 |
|
59 |
processed_demos = []
|
60 |
for demo_instance in demos:
|
61 |
-
|
62 |
-
processed_demos.append(
|
63 |
|
64 |
instance[self.demos_field] = processed_demos
|
65 |
|
|
|
19 |
class RenderTemplate(Renderer, StreamInstanceOperator):
|
20 |
template: Template
|
21 |
random_reference: bool = False
|
22 |
+
skip_rendered_instance: bool = True
|
23 |
|
24 |
def process(self, instance: Dict[str, Any], stream_name: str = None) -> Dict[str, Any]:
|
25 |
+
if self.skip_rendered_instance:
|
26 |
+
if (
|
27 |
+
"inputs" not in instance
|
28 |
+
and "outputs" not in instance
|
29 |
+
and "source" in instance
|
30 |
+
and "target" in instance
|
31 |
+
and "references" in instance
|
32 |
+
):
|
33 |
+
return instance
|
34 |
+
|
35 |
inputs = instance.pop("inputs")
|
36 |
outputs = instance.pop("outputs")
|
37 |
|
|
|
69 |
|
70 |
processed_demos = []
|
71 |
for demo_instance in demos:
|
72 |
+
demo_instance = super().process(demo_instance)
|
73 |
+
processed_demos.append(demo_instance)
|
74 |
|
75 |
instance[self.demos_field] = processed_demos
|
76 |
|