From a48a16cd5339ec2960c9a5cb0eb5a9bb07b4cc57 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 28 Oct 2024 14:13:13 -0700 Subject: [PATCH] tests --- llama_stack/providers/tests/datasetio/test_datasetio.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/tests/datasetio/test_datasetio.py b/llama_stack/providers/tests/datasetio/test_datasetio.py index adfe55896..866b1e270 100644 --- a/llama_stack/providers/tests/datasetio/test_datasetio.py +++ b/llama_stack/providers/tests/datasetio/test_datasetio.py @@ -82,7 +82,8 @@ async def register_dataset( dataset = DatasetDefWithProvider( identifier=dataset_id, - provider_id=os.environ["DATASETIO_PROVIDER_ID"] or os.environ["PROVIDER_ID"], + provider_id=os.environ.get("DATASETIO_PROVIDER_ID", None) + or os.environ["PROVIDER_ID"], url=URL( uri=test_url, ),