diff --git a/nmn/modules/executors/relate.py b/nmn/modules/executors/relate.py index 75737a578ca6e0107dea763e4a0277b46787c919..42e518d71ed9221dd7a4ff0efa70aa459a2d8c80 100644 --- a/nmn/modules/executors/relate.py +++ b/nmn/modules/executors/relate.py @@ -200,4 +200,4 @@ class ExtremeRelateExecutor(nn.Module): ) return target else: - return AttentionSet() \ No newline at end of file + return AttentionSet() diff --git a/nmn/modules/executors/same.py b/nmn/modules/executors/same.py index d4f27602fa6cc77da9b9581802e4d9ace7e4c261..6aa5a4b6a1b0cf67eec331b5d3619e3dc82700de 100644 --- a/nmn/modules/executors/same.py +++ b/nmn/modules/executors/same.py @@ -36,7 +36,11 @@ class Same(nn.Module): # assumes only one source_object # find all objects with the given concept target_set: AttentionSet = self.filter_executor( - image=image, source=filtered_set, category=category, concept=concept + image=image, + source=filtered_set, + category=category, + concept=concept, + concepts=concepts, ) target_set.set_certainties(certainty=source.get_certainty(source_object_id)) return target_set