[incubator-mxnet] 01/01: Fix check_leak_ndarray fixture for types with __hash__ NotImplementedError

[la...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

lausen pushed a commit to branch leezu-patch-2
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git

commit 0929726eda76955dc1c79d370a6d59373bff38ae
Author: Leonard Lausen <la...@amazon.com>
AuthorDate: Thu Nov 5 14:19:25 2020 -0800

    Fix check_leak_ndarray fixture for types with __hash__ NotImplementedError
    
    ndarray throws NotImplementedError on __hash__
---
 tests/python/conftest.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/conftest.py b/tests/python/conftest.py
index 4dab81d..3ac9dbb 100644
--- a/tests/python/conftest.py
+++ b/tests/python/conftest.py
@@ -58,7 +58,7 @@ def check_leak_ndarray(request):
             if element in seen:
                 return False
             seen.add(element)
-        except (TypeError, ValueError):  # unhashable
+        except (TypeError, ValueError, NotImplementedError):  # unhashable
             pass
 
         if isinstance(element, mx.nd._internal.NDArrayBase):