From 406ca72957252409a76fa0ff51b10b8891b07340 Mon Sep 17 00:00:00 2001 From: Nathan Weinberg <31703736+nathan-weinberg@users.noreply.github.com> Date: Wed, 30 Jul 2025 19:40:01 -0400 Subject: [PATCH] fix: remove redundant code from unregister_dataset (#2971) get_dataset() will raise an exception if a dataset won't be returned client handling is redundant Signed-off-by: Nathan Weinberg --- llama_stack/distribution/routing_tables/datasets.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/llama_stack/distribution/routing_tables/datasets.py b/llama_stack/distribution/routing_tables/datasets.py index 508c542a2..89da7d081 100644 --- a/llama_stack/distribution/routing_tables/datasets.py +++ b/llama_stack/distribution/routing_tables/datasets.py @@ -88,6 +88,4 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets): async def unregister_dataset(self, dataset_id: str) -> None: dataset = await self.get_dataset(dataset_id) - if dataset is None: - raise DatasetNotFoundError(dataset_id) await self.unregister_object(dataset)