From 4be63ae7033726f1b9974ab3a869638d5d69d588 Mon Sep 17 00:00:00 2001 From: Linda Ulmer Date: Fri, 11 Apr 2025 12:00:03 +0200 Subject: [PATCH] namespace comparison for numpy objects --- datamate/namespaces.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/datamate/namespaces.py b/datamate/namespaces.py index 0d64d6f..5ca4d3f 100644 --- a/datamate/namespaces.py +++ b/datamate/namespaces.py @@ -20,7 +20,7 @@ from pathlib import Path import pandas as pd - +import numpy as np __all__ = ["Namespace", "namespacify", "is_disjoint", "is_subset", "is_superset"] # -- Namespaces ---------------------------------------------------------------- @@ -542,6 +542,8 @@ def all_true(obj: Any) -> bool: return all([all_true(v.item()) for v in obj]) elif isinstance(obj, Mapping): return all([all_true(obj[k]) for k in obj]) + elif isinstance(obj, np.generic): + return bool(obj.item()) else: try: return all_true(vars(obj))