p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!14794 fix test case bug 

From: @woshixiaoli
Reviewed-by: @zichun_ye,@sunnybeike
Signed-off-by: @zichun_ye
This commit is contained in:
mindspore-ci-bot 2021-04-09 14:24:56 +08:00 committed by Gitee
parent 5c2c16d13b eaac277515
commit 018617a091
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/probability/distribution/test_categorical_gpu.py
View File

@ -14,13 +14,13 @@
# ============================================================================
"""test cases for categorical distribution"""
import pytest
import numpy as np
import mindspore.context as context
import mindspore.nn as nn
import mindspore.nn.probability.distribution as msd
from mindspore import Tensor
from mindspore import dtype as ms
import pytest
context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

2
tests/st/probability/distribution/test_cauchy_pynative.py
View File

@ -14,13 +14,13 @@
# ============================================================================
"""test cases for cauchy distribution"""
import pytest
import numpy as np
import mindspore.context as context
import mindspore.nn as nn
import mindspore.nn.probability.distribution as msd
from mindspore import Tensor
from mindspore import dtype as ms
import pytest
context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend")

2
tests/st/probability/distribution/test_gamma_pynative.py
View File

@ -14,12 +14,12 @@
# ============================================================================
"""test cases for gamma distribution"""
import pytest
import numpy as np
import mindspore.context as context
import mindspore.nn as nn
import mindspore.nn.probability.distribution as msd
from mindspore import dtype as ms
import pytest
context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend")