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

fix bug

This commit is contained in:
woshixiaoli 2021-04-08 17:47:03 +08:00
parent ac2fd8d214
commit eaac277515
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")