mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
polish code with new pre-commit rule (#2923)
This commit is contained in:
@@ -11,7 +11,7 @@
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
""" UT for air_topp_sampling kernel """
|
||||
"""UT for air_topp_sampling kernel"""
|
||||
|
||||
import unittest
|
||||
|
||||
@@ -20,7 +20,6 @@ import paddle
|
||||
|
||||
|
||||
class Test(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
"""
|
||||
Initialize.
|
||||
@@ -50,10 +49,7 @@ class Test(unittest.TestCase):
|
||||
exe.run(paddle.static.default_startup_program())
|
||||
op_out = exe.run(fetch_list=[op_out])[0]
|
||||
func_out = self.dequant_int8_test(True)
|
||||
np.testing.assert_allclose(op_out,
|
||||
func_out.numpy(),
|
||||
rtol=1e-04,
|
||||
atol=1e-04)
|
||||
np.testing.assert_allclose(op_out, func_out.numpy(), rtol=1e-04, atol=1e-04)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
Reference in New Issue
Block a user