Skip to content

Conversation

@CalebCourier
Copy link
Collaborator

@CalebCourier CalebCourier commented Sep 3, 2024

  • Adds MlFlowInstrumentor to send traces to MLFlow
  • Adds databricks extra to include mlflow module for instrumentation
  • Fixes a few edge cases in common tracing methods
  • Adds notebook to docs in integrations section to demo Guardrails w/ MLflow tracking and using a Databricks model

User can use the new instrumentation by doing this before creating their Guard:

from guardrails.integrations.databricks import MlFlowInstrumentor

MlFlowInstrumentor(experiment_name="my experiment").instrument()

@CalebCourier CalebCourier requested a review from zsimjee September 3, 2024 21:33
@CalebCourier CalebCourier marked this pull request as ready for review September 6, 2024 17:01
@CalebCourier CalebCourier requested a review from a team September 6, 2024 17:01
@zsimjee zsimjee merged commit aeb55e1 into main Sep 10, 2024
12 checks passed
@zsimjee zsimjee deleted the mlflow-integration branch September 10, 2024 02:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants