Skip to content

Commit cbecfa2

Browse files
zhoward-1claude
andcommitted
docs: replace MLflow test task with kubectl curl connectivity check
Per reviewer feedback — connectivity verification should be a simple network check, not a full Uniflow task. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 0bd074b commit cbecfa2

1 file changed

Lines changed: 6 additions & 16 deletions

File tree

docs/operator-guides/integrations/mlflow.md

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -222,25 +222,15 @@ After applying the configuration, confirm the environment variable is visible in
222222
kubectl exec -it <task-pod-name> -n <compute-namespace> -- env | grep MLFLOW
223223
```
224224

225-
You can also run a minimal test task that validates connectivity end-to-end. Submit it using your normal pipeline execution method, then check the logs for the output:
225+
You can also verify end-to-end reachability from a task pod by running a connectivity check against the MLflow health endpoint:
226226

227-
```python
228-
import os
229-
import mlflow
230-
import michelangelo.uniflow.core as uniflow
231-
from michelangelo.uniflow.plugins.ray import RayTask
232-
233-
@uniflow.task(config=RayTask(head_cpu=1, head_memory="1Gi"))
234-
def check_mlflow_connection():
235-
uri = os.environ.get("MLFLOW_TRACKING_URI", "NOT SET")
236-
print(f"MLflow tracking URI: {uri}")
237-
if uri == "NOT SET":
238-
raise ValueError("MLFLOW_TRACKING_URI is not set")
239-
client = mlflow.MlflowClient()
240-
experiments = client.search_experiments()
241-
print(f"Reachable — found {len(experiments)} experiment(s)")
227+
```bash
228+
kubectl exec -it <task-pod-name> -n <compute-namespace> -- \
229+
curl -sv http://mlflow.example.com:5000/health
242230
```
243231

232+
A `200 OK` response confirms both the environment variable injection and network reachability are working correctly.
233+
244234
---
245235

246236
## Troubleshooting

0 commit comments

Comments
 (0)