Update workflows for TensorFlow/Keras 2.14

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Nov 1, 2023 · cf8295e · cf8295e
1 parent 9614bb6
commit cf8295e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/art/defences/detector/poison/clustering_analyzer.py b/art/defences/detector/poison/clustering_analyzer.py
@@ -235,7 +235,7 @@ def analyze_by_relative_size(
             for c_id in clean_clusters[0]:
                 summary_poison_clusters[i][c_id] = 0
 
-            assigned_clean = self.assign_class(clusters, clean_clusters, poison_clusters)
+            assigned_clean = self.assign_class(clusters, clean_clusters[0], poison_clusters[0])
             all_assigned_clean.append(assigned_clean)
 
             # Generate report for this class:
@@ -328,7 +328,7 @@ def analyze_by_silhouette_score(
                     logger.info("computed silhouette score: %s", silhouette_avg)
                     dict_i.update(suspicious=True)
                 else:
-                    poison_clusters = [[]]
+                    poison_clusters = (np.array([[]]), )
                     clean_clusters = np.where(percentages >= 0)
                     dict_i.update(suspicious=False)
             else:
@@ -342,7 +342,7 @@ def analyze_by_silhouette_score(
             for c_id in clean_clusters[0]:
                 summary_poison_clusters[i][c_id] = 0
 
-            assigned_clean = self.assign_class(clusters, clean_clusters, poison_clusters)
+            assigned_clean = self.assign_class(clusters, clean_clusters[0], poison_clusters[0])
             all_assigned_clean.append(assigned_clean)
             report.update(report_class)