Skip to content
This repository was archived by the owner on Jul 19, 2023. It is now read-only.

Commit 33fe5d3

Browse files
Added integration tests for "delete" verb (#74)
* Created delete function * Created verify_delete method * Added Training, HPO and Batch delete tests * Fixed incorrect parameter * Create model before running batch * Fixed test directories * Use yq to modify model name * Added YQ apt repository * Reformatted integration test yaml * Reformatted sample test yaml * Removed duplicate configuration settings * Refactored integration and canary scripts * Increased timeout and refactored timeout code * Added SM delete checks * Renamed wait_for_crd_status * Refactored tests into their own files * Moved all run_test functionality into logical files * Formatting changes * Cast CRD type to lowercase * Added TODO in create_tests import
1 parent 6f2d387 commit 33fe5d3

33 files changed

+1375
-1248
lines changed

samples/efs-xgboost-mnist-trainingjob.yaml

Lines changed: 56 additions & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -3,59 +3,59 @@ kind: TrainingJob
33
metadata:
44
name: efs-xgboost-mnist
55
spec:
6-
hyperParameters:
7-
- name: max_depth
8-
value: "5"
9-
- name: eta
10-
value: "0.2"
11-
- name: gamma
12-
value: "4"
13-
- name: min_child_weight
14-
value: "6"
15-
- name: silent
16-
value: "0"
17-
- name: objective
18-
value: multi:softmax
19-
- name: num_class
20-
value: "10"
21-
- name: num_round
22-
value: "10"
23-
algorithmSpecification:
24-
trainingImage: 433757028032.dkr.ecr.us-west-2.amazonaws.com/xgboost:1
25-
trainingInputMode: File
26-
roleArn: arn:aws:iam::123456789012:role/service-role/AmazonSageMaker-ExecutionRole
27-
region: us-west-2
28-
outputDataConfig:
29-
s3OutputPath: s3://my-bucket/xgboost
30-
resourceConfig:
31-
instanceCount: 1
32-
instanceType: ml.m4.xlarge
33-
volumeSizeInGB: 5
34-
stoppingCondition:
35-
maxRuntimeInSeconds: 86400
36-
inputDataConfig:
37-
- channelName: train
38-
dataSource:
39-
fileSystemDataSource:
40-
fileSystemType: EFS
41-
fileSystemAccessMode: ro
42-
fileSystemId: fs-e2466d9b
43-
directoryPath: /xgboost-mnist/train
44-
contentType: text/csv
45-
compressionType: None
46-
- channelName: validation
47-
dataSource:
48-
fileSystemDataSource:
49-
fileSystemType: EFS
50-
fileSystemAccessMode: ro
51-
fileSystemId: fs-e2466d9b
52-
directoryPath: /xgboost-mnist/validation
53-
contentType: text/csv
54-
compressionType: None
55-
vpcConfig:
56-
securityGroupIds:
57-
- sg-4b23c421
58-
subnets:
59-
- subnet-ae0a65c6
60-
- subnet-2af0fa67
61-
- subnet-dddb7fa7
6+
hyperParameters:
7+
- name: max_depth
8+
value: "5"
9+
- name: eta
10+
value: "0.2"
11+
- name: gamma
12+
value: "4"
13+
- name: min_child_weight
14+
value: "6"
15+
- name: silent
16+
value: "0"
17+
- name: objective
18+
value: multi:softmax
19+
- name: num_class
20+
value: "10"
21+
- name: num_round
22+
value: "10"
23+
algorithmSpecification:
24+
trainingImage: 433757028032.dkr.ecr.us-west-2.amazonaws.com/xgboost:1
25+
trainingInputMode: File
26+
roleArn: arn:aws:iam::123456789012:role/service-role/AmazonSageMaker-ExecutionRole
27+
region: us-west-2
28+
outputDataConfig:
29+
s3OutputPath: https://s3-us-west-2.amazonaws.com/my-bucket/xgboost
30+
resourceConfig:
31+
instanceCount: 1
32+
instanceType: ml.m4.xlarge
33+
volumeSizeInGB: 5
34+
stoppingCondition:
35+
maxRuntimeInSeconds: 86400
36+
inputDataConfig:
37+
- channelName: train
38+
dataSource:
39+
fileSystemDataSource:
40+
fileSystemType: EFS
41+
fileSystemAccessMode: ro
42+
fileSystemId: fs-e2466d9b
43+
directoryPath: /xgboost-mnist/train
44+
contentType: text/csv
45+
compressionType: None
46+
- channelName: validation
47+
dataSource:
48+
fileSystemDataSource:
49+
fileSystemType: EFS
50+
fileSystemAccessMode: ro
51+
fileSystemId: fs-e2466d9b
52+
directoryPath: /xgboost-mnist/validation
53+
contentType: text/csv
54+
compressionType: None
55+
vpcConfig:
56+
securityGroupIds:
57+
- sg-4b23c421
58+
subnets:
59+
- subnet-ae0a65c6
60+
- subnet-2af0fa67
61+
- subnet-dddb7fa7

samples/fsx-xgboost-mnist-trainingjob.yaml

Lines changed: 56 additions & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -3,59 +3,59 @@ kind: TrainingJob
33
metadata:
44
name: fsx-xgboost-mnist
55
spec:
6-
hyperParameters:
7-
- name: max_depth
8-
value: "5"
9-
- name: eta
10-
value: "0.2"
11-
- name: gamma
12-
value: "4"
13-
- name: min_child_weight
14-
value: "6"
15-
- name: silent
16-
value: "0"
17-
- name: objective
18-
value: multi:softmax
19-
- name: num_class
20-
value: "10"
21-
- name: num_round
22-
value: "10"
23-
algorithmSpecification:
24-
trainingImage: 433757028032.dkr.ecr.us-west-2.amazonaws.com/xgboost:1
25-
trainingInputMode: File
26-
roleArn: arn:aws:iam::123456789012:role/service-role/AmazonSageMaker-ExecutionRole
27-
region: us-west-2
28-
outputDataConfig:
29-
s3OutputPath: s3://my-bucket/xgboost
30-
resourceConfig:
31-
instanceCount: 1
32-
instanceType: ml.m4.xlarge
33-
volumeSizeInGB: 5
34-
stoppingCondition:
35-
maxRuntimeInSeconds: 86400
36-
inputDataConfig:
37-
- channelName: train
38-
dataSource:
39-
fileSystemDataSource:
40-
fileSystemType: FSxLustre
41-
fileSystemAccessMode: ro
42-
fileSystemId: fs-0d0f8e3f721664c98
43-
directoryPath: /fsx/sagemaker/xgboost-mnist
44-
contentType: text/csv
45-
compressionType: None
46-
- channelName: validation
47-
dataSource:
48-
fileSystemDataSource:
49-
fileSystemType: FSxLustre
50-
fileSystemAccessMode: ro
51-
fileSystemId: fs-0d0f8e3f721664c98
52-
directoryPath: /fsx/sagemaker/xgboost-mnist
53-
contentType: text/csv
54-
compressionType: None
55-
vpcConfig:
56-
securityGroupIds:
57-
- sg-4b23c421
58-
subnets:
59-
- subnet-ae0a65c6
60-
- subnet-2af0fa67
61-
- subnet-dddb7fa7
6+
hyperParameters:
7+
- name: max_depth
8+
value: "5"
9+
- name: eta
10+
value: "0.2"
11+
- name: gamma
12+
value: "4"
13+
- name: min_child_weight
14+
value: "6"
15+
- name: silent
16+
value: "0"
17+
- name: objective
18+
value: multi:softmax
19+
- name: num_class
20+
value: "10"
21+
- name: num_round
22+
value: "10"
23+
algorithmSpecification:
24+
trainingImage: 433757028032.dkr.ecr.us-west-2.amazonaws.com/xgboost:1
25+
trainingInputMode: File
26+
roleArn: arn:aws:iam::123456789012:role/service-role/AmazonSageMaker-ExecutionRole
27+
region: us-west-2
28+
outputDataConfig:
29+
s3OutputPath: s3://my-bucket/xgboost
30+
resourceConfig:
31+
instanceCount: 1
32+
instanceType: ml.m4.xlarge
33+
volumeSizeInGB: 5
34+
stoppingCondition:
35+
maxRuntimeInSeconds: 86400
36+
inputDataConfig:
37+
- channelName: train
38+
dataSource:
39+
fileSystemDataSource:
40+
fileSystemType: FSxLustre
41+
fileSystemAccessMode: ro
42+
fileSystemId: fs-0d0f8e3f721664c98
43+
directoryPath: /fsx/sagemaker/xgboost-mnist
44+
contentType: text/csv
45+
compressionType: None
46+
- channelName: validation
47+
dataSource:
48+
fileSystemDataSource:
49+
fileSystemType: FSxLustre
50+
fileSystemAccessMode: ro
51+
fileSystemId: fs-0d0f8e3f721664c98
52+
directoryPath: /fsx/sagemaker/xgboost-mnist
53+
contentType: text/csv
54+
compressionType: None
55+
vpcConfig:
56+
securityGroupIds:
57+
- sg-4b23c421
58+
subnets:
59+
- subnet-ae0a65c6
60+
- subnet-2af0fa67
61+
- subnet-dddb7fa7

0 commit comments

Comments
 (0)