fix: dryrun

HerodotusDev · Jul 22, 2024 · e1c6c2c · e1c6c2c
1 parent fd9d681
commit e1c6c2c
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tools/py/schema.py b/tools/py/schema.py
@@ -104,6 +104,16 @@ class Module(ValidatedMarshmallowDataclass):
     module_class: CompiledClass
 
 
+@marshmallow_dataclass.dataclass(frozen=True)
+class DryRunModule(ValidatedMarshmallowDataclass):
+    inputs: List[int] = field(
+        metadata=additional_metadata(
+            marshmallow_field=mfields.List(IntAsHex(), required=True)
+        )
+    )
+    module_class: CompiledClass
+
+
 @marshmallow_dataclass.dataclass(frozen=True)
 class DatalakeTask:
     datalake: Datalake
@@ -139,8 +149,8 @@ class HDPInput(ValidatedMarshmallowDataclass):
 @marshmallow_dataclass.dataclass(frozen=True)
 class HDPDryRunInput(ValidatedMarshmallowDataclass):
     dry_run_output_path: str
-    modules: List[Module] = field(
+    modules: List[DryRunModule] = field(
         metadata=additional_metadata(
-            marshmallow_field=mfields.List(mfields.Nested(Module.Schema))
+            marshmallow_field=mfields.List(mfields.Nested(DryRunModule.Schema))
         )
     )