Multiple string profiling now respects grouped queries (for non-WGS)
All checks were successful
automlst.engine/pipeline/head This commit looks good

This commit is contained in:
Harrison Deng 2025-02-18 15:34:18 +00:00
parent ddf9cde175
commit 5a03c7e8d8

View File

@ -124,13 +124,17 @@ class RemoteBIGSdbMLSTProfiler(BIGSdbMLSTProfiler):
async def profile_multiple_strings(self, query_named_string_groups: AsyncIterable[Iterable[NamedString]], stop_on_fail: bool = False) -> AsyncGenerator[NamedMLSTProfile, Any]:
async for named_strings in query_named_string_groups:
names: list[str] = list()
sequences: list[str] = list()
for named_string in named_strings:
names.append(named_string.name)
sequences.append(named_string.sequence)
try:
yield NamedMLSTProfile(named_string.name, (await self.profile_string([named_string.sequence])))
yield NamedMLSTProfile("-".join(names), (await self.profile_string(sequences)))
except NoBIGSdbMatchesException as e:
if stop_on_fail:
raise e
yield NamedMLSTProfile(named_string.name, None)
yield NamedMLSTProfile("-".join(names), None)
async def close(self):
await self._http_client.close()