From 2ed8a4dbb832b2f81873faf6b025e180b5b764d7 Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Tue, 19 Nov 2024 16:35:30 -0500 Subject: [PATCH] rm comment --- tests/callbacks/test_callbacks.py | 1 - tests/callbacks/test_loggers_across_callbacks.py | 1 - tests/loggers/test_mlflow_logger.py | 1 - 3 files changed, 3 deletions(-) diff --git a/tests/callbacks/test_callbacks.py b/tests/callbacks/test_callbacks.py index 4cdf286425..30984bb48a 100644 --- a/tests/callbacks/test_callbacks.py +++ b/tests/callbacks/test_callbacks.py @@ -23,7 +23,6 @@ @pytest.fixture(autouse=True) def setup_mlflow_tracking(monkeypatch, tmp_path): mlflow = pytest.importorskip('mlflow') - # Use a temporary directory instead of 'databricks' tracking_uri = str(tmp_path / 'mlruns') monkeypatch.setenv(mlflow.environment_variables.MLFLOW_TRACKING_URI.name, tracking_uri) os.makedirs(tracking_uri, exist_ok=True) diff --git a/tests/callbacks/test_loggers_across_callbacks.py b/tests/callbacks/test_loggers_across_callbacks.py index 828d4429f2..3d69d3206d 100644 --- a/tests/callbacks/test_loggers_across_callbacks.py +++ b/tests/callbacks/test_loggers_across_callbacks.py @@ -20,7 +20,6 @@ @pytest.fixture(autouse=True) def setup_mlflow_tracking(monkeypatch, tmp_path): mlflow = pytest.importorskip('mlflow') - # Use a temporary directory instead of 'databricks' tracking_uri = str(tmp_path / 'mlruns') monkeypatch.setenv(mlflow.environment_variables.MLFLOW_TRACKING_URI.name, tracking_uri) os.makedirs(tracking_uri, exist_ok=True) diff --git a/tests/loggers/test_mlflow_logger.py b/tests/loggers/test_mlflow_logger.py index 65a7f53ca7..5eb3a5a2e7 100644 --- a/tests/loggers/test_mlflow_logger.py +++ b/tests/loggers/test_mlflow_logger.py @@ -28,7 +28,6 @@ @pytest.fixture(autouse=True) def setup_mlflow_tracking(monkeypatch, tmp_path): mlflow = pytest.importorskip('mlflow') - # Use a temporary directory instead of 'databricks' tracking_uri = str(tmp_path / 'mlruns') monkeypatch.setenv(mlflow.environment_variables.MLFLOW_TRACKING_URI.name, tracking_uri) os.makedirs(tracking_uri, exist_ok=True)