提交 2c4d1afe 编写于 作者: M Megvii Engine Team

test(mge): move `NUMBER` config to pytest.init

GitOrigin-RevId: 1d82209c409ccf884ed9f7917a51ea4da62ec00b
上级 f33a92d6
......@@ -190,7 +190,6 @@ def sqrt(inp: Tensor) -> Tensor:
Outputs:
.. testoutput::
:options: +NUMBER
[[0. 1. 1.4142]
[1.7321 2. 2.2361 ]]
......
......@@ -636,7 +636,6 @@ def interpolate(
Outputs:
.. testoutput::
:options: +NUMBER
[[[[1. 1.25 1.75 2. ]
[1.5 1.75 2.25 2.5 ]
......
......@@ -39,7 +39,6 @@ def argsort(inp: Tensor, descending: bool = False) -> Tuple[Tensor, Tensor]:
Outputs:
.. testoutput::
:options: +NUMBER
[1. 2.] [0 1]
......@@ -93,7 +92,6 @@ def top_k(
Outputs:
.. testoutput::
:options: +NUMBER
[1. 2. 3. 4. 5.] [7 0 6 1 5]
......
......@@ -50,7 +50,6 @@ def accuracy(logits: Tensor, target: Tensor, topk: Union[int, Iterable[int]] = 1
Outputs:
.. testoutput::
:options: +NUMBER
[0.] [0.375]
"""
......
......@@ -218,7 +218,6 @@ class LeakyReLU(Module):
Outputs:
.. testoutput::
:options: +NUMBER
[-0.08 -0.12 6. 10. ]
......
......@@ -138,7 +138,6 @@ class Embedding(Module):
Outputs:
.. testoutput::
:options: +NUMBER
[[[1.2 2.3 3.4 4.5 5.6]
[0.1 1.1 2.1 3.1 4.1]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册