Skip to content

Allow the load test script to use a csv of inputs#440

Merged
seanshi-scale merged 17 commits intomainfrom
seanshi/load-script-use-production-data
Feb 27, 2024
Merged

Allow the load test script to use a csv of inputs#440
seanshi-scale merged 17 commits intomainfrom
seanshi/load-script-use-production-data

Conversation

@seanshi-scale
Copy link
Copy Markdown
Contributor

@seanshi-scale seanshi-scale commented Feb 21, 2024

Pull Request Summary

Allow passing in a json-formatted list of output token counts

Test Plan and Usage Guide

Ran the script

@seanshi-scale seanshi-scale changed the title prepare allowing a csv input Allow the load test script to use a csv of inputs Feb 21, 2024
# Only supports csvs for now
if re.match(r".*\.csv$", input_file):
with open(input_file, "r", newline="") as file:
reader = csv.reader(file)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: try catch error

@seanshi-scale seanshi-scale self-assigned this Feb 26, 2024
@seanshi-scale seanshi-scale marked this pull request as ready for review February 26, 2024 20:42
@yunfeng-scale
Copy link
Copy Markdown
Contributor

what's an example distribution?

@seanshi-scale
Copy link
Copy Markdown
Contributor Author

what's an example distribution?

the file would contain some text like [1000, 1001, 1010, 1024, 512, 512, 513] for instance

@seanshi-scale seanshi-scale merged commit 9a4e2e5 into main Feb 27, 2024
@seanshi-scale seanshi-scale deleted the seanshi/load-script-use-production-data branch February 27, 2024 01:16
@yunfeng-scale yunfeng-scale mentioned this pull request Mar 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants