Skip to content

Fix inference unit test import error catching#2024

Merged
mrwyattii merged 9 commits intodeepspeedai:masterfrom
mrwyattii:pytest-inference-tests-import-fix
Jun 21, 2022
Merged

Fix inference unit test import error catching#2024
mrwyattii merged 9 commits intodeepspeedai:masterfrom
mrwyattii:pytest-inference-tests-import-fix

Conversation

@mrwyattii
Copy link
Contributor

@mrwyattii mrwyattii commented Jun 16, 2022

Fixes the error we see here, when not running inference tests: https://github.com/microsoft/DeepSpeed/runs/6912710983?check_suite_focus=true#step:7:1409

This change also addresses a bug where an import error when -m "inference" is passed to pytest and there is an import error, all tests are skipped, and the test succeeds

Also adds a check at the beginning of testing for expected torch and cuda versions (this is in response to seeing that pip will sometimes upgrade torch when setting up the environment)

@mrwyattii mrwyattii changed the title Catch Assertion Error for inference unit test imports Fix inference unit test import error catching Jun 16, 2022
@mrwyattii mrwyattii merged commit ec1ec20 into deepspeedai:master Jun 21, 2022
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