feature/datasets - add scavenger example dataset #52
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Dataset loading functions updates:
cpm/datasets/base.py
: Modified theload_bandit_data
function to update column names and added a new functionload_risky_choices
to load the risky choices dataset.Test cases for dataset loading functions:
test/datasets/test_datasets.py
: Added test cases forload_csv
,load_bandit_data
, andload_risky_choices
functions usingpytest
andunittest.mock
for mocking file existence andpandas.read_csv
behavior.New dataset addition:
cpm/datasets/data/risky_choices.csv
: Added a new CSV file containing the risky choices dataset.