Fix dataset retrieval query for `Geno` traits

Issue: https://github.com/genenetwork/gn-gemtext-threads/blob/main/topics/gn1-migration-to-gn2/partial-correlations.gmi

1 files changed, 3 insertions, 3 deletions

diff --git a/gn3/db/partial_correlations.py b/gn3/db/partial_correlations.py
index 0befb34..157f8ee 100644
--- a/gn3/db/partial_correlations.py
+++ b/gn3/db/partial_correlations.py
@@ -540,12 +540,12 @@ def geno_datasets_names(conn, threshold, dataset_names):
         "WHERE "
         "public > %s "
         "AND "
-        "(Name = ({names}) OR FullName = ({names}) OR ShortName = ({names}))")
+        "(Name IN ({names}) OR FullName IN ({names}) OR ShortName IN ({names}))")
     with conn.cursor(cursorclass=DictCursor) as cursor:
         cursor.execute(
             query.format(names=", ".join(["%s"] * len(dataset_names))),
-            (threshold,) + tuple(dataset_names))
-        return ...
+            (threshold,) + (tuple(dataset_names) * 3))
+        return {ds["dataset_name"]: ds for ds in cursor.fetchall()}
     return {}
 
 def geno_datasets_groups(conn, dataset_names):