Skip to content

GitLab

"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "01abfc873659e29a8d002f20782fa5b5e6d03f9c"
Unverified Commit d6846203 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

add version check test (#9)

parent 52c5a936
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
tests/unit/test_config.py
View file @ d6846203
...@@ -7,3 +7,9 @@ import deepspeed ...@@ -7,3 +7,9 @@ import deepspeed
def test_cuda(): def test_cuda():
assert (torch.cuda.is_available()) assert (torch.cuda.is_available())
def test_check_version():
assert hasattr(deepspeed, "__git_hash__")
assert hasattr(deepspeed, "__git_branch__")
assert hasattr(deepspeed, "__version__")
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment