Modified generator to generate plugins

Signed-off-by: saimedhi <[email protected]>
saimedhi · Mar 21, 2024 · 27bf22f · 27bf22f
1 parent 2c0cf07
commit 27bf22f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test_opensearchpy/test_server/test_plugins/test_knn.py b/test_opensearchpy/test_server/test_plugins/test_knn.py
@@ -85,11 +85,12 @@ def test_knn_model(self) -> None:
 
             # Fetch the model state
             try:
-                response = self.client.plugins.knn.get_model(model_id=model_id, filter_path='state')['state']
+                response = self.client.plugins.knn.get_model(model_id='my-model')
                 if response['state'] != 'training':
                     pass 
-            except Exception as e:
-                print(f"Error fetching knn model: {e}")
+            except Exception as error:
+                print(f"Error fetching knn model: {error}")
+                break
 
             # Sleep for 1 minute before the next attempt
             time.sleep(10)