-
Notifications
You must be signed in to change notification settings - Fork 758
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add pytest option to generate a functional report for distribution #833
Conversation
d8ec744
to
0246333
Compare
rows = [] | ||
for model in all_registered_models(): | ||
if ( | ||
"Instruct" not in model.core_model_id.value |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
model.is_instruct_model()
might be better to use for instruct models.
and for safety models, might be better to check model family and if its ModelFamily.safety
INFERENCE_API_CAPA_TEST_MAP = { | ||
"chat_completion": { | ||
"streaming": [ | ||
"test_text_chat_completion_streaming", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this implies that we need to manually add tests to this list , everytime we add new capabilities. Not ideal but something we can look into as a follow up .
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@hardikjshah
Since we want to find out the corresponding tests given an api and the capability, the other approaches are
- parse test function name
- add markers (ex. pytest.marker.structured_output) to each test
I think both would be less cleaner and therefore i prefer the current approach. Lemme know if you have other suggestions! :)
c9fc1c0
to
2ace8d2
Compare
2ace8d2
to
e9f49a1
Compare
What does this PR do?
add pytest option (
--report
) to support generating a functional report for llama stack distributionTest Plan
See a report file was generated under
./llama_stack/templates/fireworks/report.md
Sources
Please link relevant resources if necessary.
Before submitting
Pull Request section?