From 404c0b3e423491f115d4a498725d5de61d4f1ac4 Mon Sep 17 00:00:00 2001
From: kalyan <kalyan.ben10@live.com>
Date: Tue, 31 Oct 2023 08:31:09 +0530
Subject: [PATCH] revert

Signed-off-by: kalyan <kalyan.ben10@live.com>
---
 tests/ml_commons/test_model_train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/ml_commons/test_model_train.py b/tests/ml_commons/test_model_train.py
index cc1561e1d..07a0d9080 100644
--- a/tests/ml_commons/test_model_train.py
+++ b/tests/ml_commons/test_model_train.py
@@ -61,7 +61,9 @@ def iris_index():
     # without the sleep, test is failing.
     time.sleep(2)
 
-    return index_name
+    yield index_name
+
+    ml_client._client.indices.delete(index=index_name)
 
 
 def test_init():