update provider to provider_id

This commit is contained in:
Rashmi Pawar 2025-04-16 14:55:21 +05:30
parent 4491a51149
commit 00e57d693f
3 changed files with 4 additions and 4 deletions

View file

@ -60,7 +60,7 @@ class TestNvidiaDatastore(unittest.TestCase):
provider_id="",
purpose=DatasetPurpose.post_training_messages,
source=URIDataSource(uri="https://example.com/data.jsonl"),
metadata={"format": "jsonl", "description": "Test dataset description"},
metadata={"provider_id": "nvidia", "format": "jsonl", "description": "Test dataset description"},
)
self.run_async(self.adapter.register_dataset(dataset_def))