allow lookuptable to skip regex searching for exact matches

astropy · Dec 19, 2023 · 869de3d · 869de3d
1 parent 5600762
commit 869de3d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/astroquery/jplspec/lookup_table.py b/astroquery/jplspec/lookup_table.py
@@ -22,10 +22,13 @@ def find(self, s, flags):
 
         Returns
         -------
-        The list of values corresponding to the matches
+        The dictionary containing only values whose keys match the regex
 
         """
 
+        if s in self:
+            return {s: self[s]}
+
         R = re.compile(s, flags)
 
         out = {}

diff --git a/astroquery/linelists/cdms/core.py b/astroquery/linelists/cdms/core.py
@@ -375,10 +375,13 @@ def find(self, st, flags):
 
         Returns
         -------
-        The list of values corresponding to the matches
+        The dictionary containing only values whose keys match the regex
 
         """
 
+        if st in self:
+            return {st: self[st]}
+
         out = {}
 
         for kk, vv in self.items():