Add All Folders

Dagster/Check_Matching/nodes/feature_writer.py

@@ -0,0 +1,33 @@
+import json
+from core.node import Node
+from core.compatibility_result import CompatibilityResult, CompatibilityLevel
+
+
+class FeatureWriteRequirement:
+    data_type: str = "table"
+
+    def check(self, provided):
+        if provided.data_type != self.data_type:
+            return CompatibilityResult(
+                level=CompatibilityLevel.ERROR,
+                message="FeatureWriter requires feature table input"
+            )
+        return CompatibilityResult(
+            level=CompatibilityLevel.OK,
+            message="Compatible"
+        )
+
+
+class FeatureWriter(Node):
+    def __init__(self, output_path: str):
+        self.output_path = output_path
+
+        super().__init__(
+            name="FeatureWriter",
+            inputs={"features": FeatureWriteRequirement()},
+            outputs={}
+        )
+
+    def run(self, features):
+        with open(self.output_path, "w", encoding="utf-8") as f:
+            json.dump(features, f, indent=2)