quick fix

This commit is contained in:
lda committed 2025-05-01 23:23:10 +07:00
1 parent 508be518e9
commit 20412d225e
1 file changed
+2 -2
+2 -2
View File
@@ -128,9 +128,9 @@ def query_once_fetch_lda(
} }
r.execute(*execute_settings) r.execute(*execute_settings)
if isinstance(fetch_many, fn): if isinstance(fetch_many, str):
return rf[fetch_many]() return rf[fetch_many]()
if isinstance(fetch_many, Collection[fn]): if isinstance(fetch_many, Collection):
return tuple(rf[i]() for i in fetch_many) return tuple(rf[i]() for i in fetch_many)
raise AttributeError( raise AttributeError(
f"please put one of the one, many, all, warnings in ty (you put {fetch_many})", f"please put one of the one, many, all, warnings in ty (you put {fetch_many})",