Skip to content

Commit bd4583c

Browse files
committed
Remove call to internal, backwards incompatible api
1 parent 703841a commit bd4583c

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

guardrails/integrations/databricks/ml_flow_instrumentor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,7 @@ def __init__(self, experiment_name: str):
5454
settings.disable_tracing = True
5555

5656
def instrument(self):
57-
if not mlflow.tracing.provider._is_enabled():
58-
mlflow.tracing.enable()
57+
mlflow.tracing.enable()
5958
mlflow.set_experiment(self.experiment_name)
6059

6160
wrapped_guard_execute = self._instrument_guard(Guard._execute)

poetry.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ guardrails-hub-types = "^0.0.4"
5959
guardrails-api-client = "^0.3.13"
6060
diff-match-patch = "^20230430"
6161
guardrails-api = "^0.0.3"
62-
mlflow = {version = ">=2.0.1", optional = true}
62+
mlflow = {version = "^2.0.1", optional = true}
6363
uvloop = {version = "^0.20.0", optional = true}
6464
semver = "^3.0.2"
6565

0 commit comments

Comments
 (0)