From ad7040e7cde0dbc57ab8523ce5e7c37a8554d6e9 Mon Sep 17 00:00:00 2001 From: Mayur Singal <39544459+ulixius9@users.noreply.github.com> Date: Thu, 14 Apr 2022 12:59:33 +0530 Subject: [PATCH] Fix #4125: Fix sample data index error (#4126) --- ingestion/src/metadata/ingestion/source/sample_data.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/ingestion/source/sample_data.py b/ingestion/src/metadata/ingestion/source/sample_data.py index 1783d88e254..d8d28b2b954 100644 --- a/ingestion/src/metadata/ingestion/source/sample_data.py +++ b/ingestion/src/metadata/ingestion/source/sample_data.py @@ -454,10 +454,13 @@ class SampleDataSource(Source[Entity]): ) resp = self.metadata.list_entities(entity=User, limit=5) self.user_entity = resp.entities + user_entity_len = min(len(self.user_entity), 5) for table in self.tables["tables"]: try: for sql_object in table["tableQueries"]: - user_entity = self.user_entity[random.choice(range(5))] + user_entity = self.user_entity[ + random.choice(range(user_entity_len)) + ] user_dict = { "id": user_entity.id.__root__, "name": user_entity.name.__root__,