fix: corrected type dict for States in MapState (#32)

skyleague · Mar 21, 2024 · 55b8b12 · 55b8b12
1 parent f423d68
commit 55b8b12
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 deletions.
diff --git a/src/types/schemas/state-machine.schema.js b/src/types/schemas/state-machine.schema.js
diff --git a/src/types/sfn.schema.ts b/src/types/sfn.schema.ts
@@ -74,7 +74,7 @@ export const processConfig = $object({
 export const itemProcessor = $object({
     ProcessorConfig: $optional($ref(processConfig)),
     StartAt: $string(),
-    States: $ref(taskState),
+    States: $dict($ref(taskState)),
 })
 
 export const readerConfig = $object({
@@ -96,8 +96,8 @@ export const itemReader = $object({
 })
 
 export const itemBatcher = $object({
-    MaxItemsPerBatchPath: $optional($dict($string)),
-    MaxInputBytesPerBatchPath: $optional($dict($string)),
+    MaxItemsPerBatchPath: $optional($dict($number)),
+    MaxInputBytesPerBatchPath: $optional($dict($number)),
 })
 
 export const mapState = $union(

diff --git a/src/types/sfn.type.ts b/src/types/sfn.type.ts
@@ -14,17 +14,19 @@ export interface FailState {
 
 export interface ItemBatcher {
     MaxItemsPerBatchPath?: {
-        [k: string]: string | undefined
+        [k: string]: number | undefined
     }
     MaxInputBytesPerBatchPath?: {
-        [k: string]: string | undefined
+        [k: string]: number | undefined
     }
 }
 
 export interface ItemProcessor {
     ProcessorConfig?: ProcessConfig
     StartAt: string
-    States: TaskState
+    States: {
+        [k: string]: TaskState | undefined
+    }
 }
 
 export interface ItemReader {