revert

Signed-off-by: kalyan <[email protected]>
opensearch-project · Oct 31, 2023 · 404c0b3 · 404c0b3
1 parent 70f0a75
commit 404c0b3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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():