From 027742c95371a559859ad0c47377acb2f7cca2e0 Mon Sep 17 00:00:00 2001 From: Juan Mrad Date: Thu, 3 Oct 2024 12:36:49 +0100 Subject: [PATCH] [hma] Fix Broken UI due to lookup change (#1642) --- .../src/OpenMediaMatch/blueprints/development.py | 2 +- .../src/OpenMediaMatch/blueprints/matching.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/development.py b/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/development.py index 6e9abc0dd..d008bb9c1 100644 --- a/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/development.py +++ b/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/development.py @@ -53,7 +53,7 @@ def query_media(): return signal_type_to_signal_map abort(500, "Something went wrong while hashing the provided media.") - include_distance = bool(request.args.get("include_distance", False)) == True + include_distance = str_to_bool(request.args.get("include_distance", "false")) lookup_signal_func = ( lookup_signal_with_distance if include_distance else lookup_signal ) diff --git a/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/matching.py b/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/matching.py index fd518527d..42fd7efc4 100644 --- a/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/matching.py +++ b/hasher-matcher-actioner/src/OpenMediaMatch/blueprints/matching.py @@ -237,9 +237,8 @@ def lookup(signal, signal_type_name): raw_results = lookup_signal(signal, signal_type_name) storage = get_storage() current_app.logger.debug("getting bank content") - contents = storage.bank_content_get( - {cid for l in raw_results.values() for cid in l} - ) + current_app.logger.debug(raw_results) + contents = storage.bank_content_get(raw_results) enabled = [c for c in contents if c.enabled] current_app.logger.debug( "lookup matches %d content ids (%d enabled)", len(contents), len(enabled)