[hma] Fix Broken UI due to lookup change (#1642)

facebook · Oct 3, 2024 · 027742c · 027742c
1 parent 806f30e
commit 027742c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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)