Skip to content

Commit f177861

Browse files
committed
remove _is_enabled mock
1 parent bd4583c commit f177861

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

tests/unit_tests/integrations/databricks/test_ml_flow_instrumentor.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,6 @@ def test__init__(self):
3737
assert settings.disable_tracing is True
3838

3939
def test_instrument(self, mocker):
40-
mock_is_enabled = mocker.patch(
41-
"guardrails.integrations.databricks.ml_flow_instrumentor.mlflow.tracing.provider._is_enabled",
42-
return_value=False,
43-
)
4440
mock_enable = mocker.patch(
4541
"guardrails.integrations.databricks.ml_flow_instrumentor.mlflow.tracing.enable"
4642
)
@@ -115,7 +111,6 @@ def test_instrument(self, mocker):
115111

116112
m.instrument()
117113

118-
mock_is_enabled.assert_called_once()
119114
mock_enable.assert_called_once()
120115
mock_set_experiment.assert_called_once_with("mock experiment")
121116

0 commit comments

Comments
 (0)