(
inference_client: InferenceRouteClient,
)
| 106 | |
| 107 | |
| 108 | def _current_cluster_inference( |
| 109 | inference_client: InferenceRouteClient, |
| 110 | ) -> ClusterInferenceConfig | None: |
| 111 | try: |
| 112 | return inference_client.get_cluster() |
| 113 | except grpc.RpcError as exc: |
| 114 | if exc.code() == grpc.StatusCode.NOT_FOUND: |
| 115 | return None |
| 116 | raise |
| 117 | |
| 118 | |
| 119 | def _restore_cluster_inference( |
no test coverage detected