Skip to content

Commit d669c9e

Browse files
authored
Fix for name of collection being a tuple (aws#242)
* Fix tuple going in as name of coll * Add xgboost to CI * Install xgboost
1 parent 5c29851 commit d669c9e

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

config/buildspec.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ env:
66
run_pytest_pytorch: "enable"
77
run_pytest_mxnet: "enable"
88
run_pytest_tensorflow: "enable"
9+
run_pytest_xgboost: "enable"
910
phases:
1011
install:
1112
commands:
@@ -15,7 +16,7 @@ phases:
1516
- sudo apt-get update
1617
- sudo apt-get install unzip
1718
- 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
1920
- pip uninstall -y boto3 && pip uninstall -y aiobotocore && pip uninstall -y botocore
2021

2122
pre_build:

tornasole/core/collection.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,10 @@ class CollectionKeys:
2424
HISTOGRAMS = 'histograms'
2525

2626
#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"
3131
AVERAGE_SHAP = "average_shap"
3232

3333

0 commit comments

Comments
 (0)