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 |
|