Style fixes and exceptions

Signed-off-by: Kevin Eykholt <kheykholt@gmail.com>
Trusted-AI · Nov 9, 2021 · b9bc8a1 · b9bc8a1
1 parent a96b4a6
commit b9bc8a1
Show file tree

Hide file tree

Showing 2 changed files with 727 additions and 837 deletions.
diff --git a/art/estimators/classification/deep_partition_ensemble.py b/art/estimators/classification/deep_partition_ensemble.py
@@ -95,10 +95,12 @@ def __init__(
         )
 
         if hash_function is None:
-
             def hash_function(x):
                 return int(np.sum(x)) % ensemble_size
 
+        if not isinstance(hash_function, Callable):
+            raise ValueError("hash_function must be callable")
+
         self.hash_function = hash_function
         self.ensemble_size = ensemble_size