File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff line change 6
6
run_pytest_pytorch : " enable"
7
7
run_pytest_mxnet : " enable"
8
8
run_pytest_tensorflow : " enable"
9
+ run_pytest_xgboost : " enable"
9
10
phases :
10
11
install :
11
12
commands :
@@ -15,7 +16,7 @@ phases:
15
16
- sudo apt-get update
16
17
- sudo apt-get install unzip
17
18
- cd $CODEBUILD_SRC_DIR && chmod +x config/protoc_downloader.sh && ./config/protoc_downloader.sh
18
- - pip install pytest wheel pyYaml pytest-html tensorflow mxnet torch
19
+ - pip install pytest wheel pyYaml pytest-html tensorflow mxnet torch xgboost
19
20
- pip uninstall -y boto3 && pip uninstall -y aiobotocore && pip uninstall -y botocore
20
21
21
22
pre_build :
Original file line number Diff line number Diff line change @@ -24,10 +24,10 @@ class CollectionKeys:
24
24
HISTOGRAMS = 'histograms'
25
25
26
26
#XGBOOST
27
- METRIC = "metric" ,
28
- PREDICTIONS = "predictions" ,
29
- LABELS = "labels" ,
30
- FEATURE_IMPORTANCE = "feature_importance" ,
27
+ METRIC = "metric"
28
+ PREDICTIONS = "predictions"
29
+ LABELS = "labels"
30
+ FEATURE_IMPORTANCE = "feature_importance"
31
31
AVERAGE_SHAP = "average_shap"
32
32
33
33
You can’t perform that action at this time.
0 commit comments