From 6305131317b917147cc2825f46304541d122f552 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Zientkiewicz?= Date: Thu, 12 Sep 2024 10:52:13 +0200 Subject: [PATCH] Fix test. Remove unused mxnet test. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Michał Zientkiewicz --- dali/test/python/test_pipeline_debug.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/dali/test/python/test_pipeline_debug.py b/dali/test/python/test_pipeline_debug.py index 762652ed82e..31dbef3a9c5 100644 --- a/dali/test/python/test_pipeline_debug.py +++ b/dali/test/python/test_pipeline_debug.py @@ -1,4 +1,4 @@ -# Copyright (c) 2021-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright (c) 2021-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -168,13 +168,6 @@ def test_injection_numpy(): _test_injection("cpu", "numpy array", lambda xs: [np.array(x) for x in xs]) -@attr("mxnet") -def test_injection_mxnet(): - import mxnet - - _test_injection("cpu", "mxnet array", lambda xs: [mxnet.nd.array(x, dtype="uint8") for x in xs]) - - @attr("pytorch") def test_injection_torch(): import torch @@ -544,7 +537,7 @@ def cpu_after_gpu_pipeline(): return labels, output -@raises(RuntimeError, glob="Cannot call * operator * with * input *") +@raises(RuntimeError, glob='incompatible device "gpu"') def test_cpu_operator_after_gpu(): pipe = cpu_after_gpu_pipeline() pipe.build()