vnikedsst

LLNL · Jun 9, 2023 · 4554cee · 4554cee
1 parent f47a65a
commit 4554cee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/thicket/tests/test_query.py b/thicket/tests/test_query.py
@@ -58,7 +58,7 @@ def test_new_style_query_base(rajaperf_basecuda_xl_cali):
         )
     )
 
-    check_query(th, hnids, query)
+    check_query(th, hnids, query, multi_index_mode="off")
 
 
 def test_new_style_query_object(rajaperf_basecuda_xl_cali):
@@ -100,4 +100,4 @@ def test_old_style_query(rajaperf_basecuda_xl_cali):
         )
     )
 
-    check_query(th, hnids, query)
+    check_query(th, hnids, query, multi_index_mode="off")
diff --git a/thicket/thicket.py b/thicket/thicket.py
@@ -1114,7 +1114,7 @@ def query(
         Returns:
             (Thicket): a new Thicket object containing the data that matches the query
         """
-        if not is_thicket_query(query_obj):
+        if not is_thicket_query(query_obj) or not isinstance(query_obj, (list, str)):
             raise TypeError(
                 "Input to 'query' must be a Hatchet query (i.e., list, str, or new- or old-style query object)"
             )