wehub-resource-sync
30 lines
776 B
Python
30 lines
776 B
Python
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
"MLFLOW_SKINNY" not in os.environ, reason="This test is only valid for the skinny client"
|
|
)
|
|
def test_fails_import_sqlalchemy():
|
|
import mlflow # noqa: F401
|
|
|
|
with pytest.raises(ImportError, match="sqlalchemy"):
|
|
import sqlalchemy # noqa: F401
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
"MLFLOW_SKINNY" not in os.environ, reason="This test is only valid for the skinny client"
|
|
)
|
|
def test_skinny_client_without_importing_sqlalchemy() -> None:
|
|
from mlflow.tracking.client import MlflowClient
|
|
|
|
client = MlflowClient(
|
|
tracking_uri="databricks",
|
|
registry_uri="databricks",
|
|
)
|
|
client._tracking_client.store
|
|
client._get_registry_client().store
|
|
assert "sqlalchemy" not in sys.modules
|