Adding in grpc code

triton-inference-server · Feb 20, 2024 · 68aa75f · 68aa75f
1 parent ee06d39
commit 68aa75f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/model_analyzer/triton/client/grpc_client.py b/model_analyzer/triton/client/grpc_client.py
@@ -84,3 +84,9 @@ def get_model_repository_index(self):
         Returns the JSON dict holding the model repository index.
         """
         return self._client.get_model_repository_index(as_json=True)["models"]
+
+    def is_model_ready(self, model_name: str) -> bool:
+        """
+        Returns true if the model is loaded on the server
+        """
+        return self._client.is_model_ready(model_name)
diff --git a/model_analyzer/triton/client/http_client.py b/model_analyzer/triton/client/http_client.py
@@ -104,6 +104,4 @@ def is_model_ready(self, model_name: str) -> bool:
         """
         Returns true if the model is loaded on the server
         """
-        foo = self._client.is_model_ready(model_name)
-
-        return foo
+        return self._client.is_model_ready(model_name)