Implement list

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
mlflow · Nov 16, 2023 · 33f082d · 33f082d
1 parent dd1768c
commit 33f082d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/examples/deployments/databricks.py b/examples/deployments/databricks.py
@@ -34,6 +34,7 @@ def main():
         },
     )
     try:
+        print(json.dumps(client.list_endpoints()["endpoints"][:3], indent=2))
         print(json.dumps(client.get_endpoint(endpoint=name), indent=2))
         print(
             json.dumps(

diff --git a/mlflow/deployments/databricks/__init__.py b/mlflow/deployments/databricks/__init__.py
@@ -83,7 +83,7 @@ def delete_endpoint(self, endpoint):
         return self._call_endpoint("DELETE", route=endpoint)
 
     def list_endpoints(self):
-        raise NotImplementedError("TODO")
+        return self._call_endpoint("GET")
 
     def get_endpoint(self, endpoint):
         return self._call_endpoint("GET", route=endpoint)