fix error with bedrock-finetuning module deployment (#108)

awslabs · Jun 4, 2024 · cb0f731 · cb0f731
1 parent 932aade
commit cb0f731
Show file tree

Hide file tree

Showing 4 changed files with 1,219 additions and 13 deletions.
diff --git a/modules/fmops/bedrock-finetuning/deployspec.yaml b/modules/fmops/bedrock-finetuning/deployspec.yaml
@@ -8,7 +8,7 @@ deploy:
     build:
       commands:
         - env
-        - cdk deploy --require-approval never --progress events --app "npx ts-node --prefer-ts-exts bin/mlops_modules_development.ts"
+        - cdk deploy --require-approval never --progress events --app "npx ts-node --prefer-ts-exts bin/bedrock-finetuning.ts"
 destroy:
   phases:
     install:
@@ -17,4 +17,4 @@ destroy:
         - npm install
     build:
       commands:
-        - cdk destroy --force --app "npx ts-node --prefer-ts-exts index.ts"
+        - cdk destroy --force --app "npx ts-node --prefer-ts-exts bin/bedrock-finetuning.ts"
diff --git a/modules/fmops/bedrock-finetuning/lib/bedrock-finetuning-stack.ts b/modules/fmops/bedrock-finetuning/lib/bedrock-finetuning-stack.ts
@@ -169,7 +169,7 @@ export class AmazonBedrockFinetuningStack extends cdk.Stack {
       new sfn.Succeed(this, "Succeed"),
     );
     const stateMachine = new sfn.StateMachine(this, "MyStateMachine", {
-      definition: definition,
+      definitionBody: sfn.DefinitionBody.fromChainable(definition),
       timeout: cdk.Duration.minutes(5),
       stateMachineName: "BedrockFinetuning",
       tracingEnabled: true,