From b92f2c4ed3484789f3faf67e40ef82cab75bbd13 Mon Sep 17 00:00:00 2001 From: luandalmazo Date: Thu, 20 Aug 2026 09:40:52 -0300 Subject: [PATCH] moved MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR --- datamint/mlflow/__init__.py | 3 +++ .../06_end_to_end/full_3d/01_synapse_unetrpp.ipynb | 3 --- .../06_end_to_end/full_3d/02_synapse_nnunet.ipynb | 3 --- .../slice_based/01_fracatlas_classification.ipynb | 3 --- .../slice_based/02_busi_segmentation.ipynb | 12 ------------ .../slice_based/03_bccd_detection.ipynb | 3 --- 6 files changed, 3 insertions(+), 24 deletions(-) diff --git a/datamint/mlflow/__init__.py b/datamint/mlflow/__init__.py index 3989739e..94ddc218 100644 --- a/datamint/mlflow/__init__.py +++ b/datamint/mlflow/__init__.py @@ -1,5 +1,6 @@ # Monkey patch mlflow.tracking._tracking_service.utils.get_tracking_uri import logging +import os from functools import wraps from typing import TYPE_CHECKING @@ -9,6 +10,8 @@ _LOGGER = logging.getLogger(__name__) +os.environ.setdefault('MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR', 'false') + # Store reference to original function _original_get_tracking_uri = mlflow_utils.get_tracking_uri _SETUP_CALLED_SUCCESSFULLY = False diff --git a/notebooks/06_end_to_end/full_3d/01_synapse_unetrpp.ipynb b/notebooks/06_end_to_end/full_3d/01_synapse_unetrpp.ipynb index f408f47a..a6a51faf 100644 --- a/notebooks/06_end_to_end/full_3d/01_synapse_unetrpp.ipynb +++ b/notebooks/06_end_to_end/full_3d/01_synapse_unetrpp.ipynb @@ -159,9 +159,6 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", - "\n", - "os.environ['MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR'] = 'false'\n", "\n", "from datamint.lightning import UNETRPPTrainer\n", "\n", diff --git a/notebooks/06_end_to_end/full_3d/02_synapse_nnunet.ipynb b/notebooks/06_end_to_end/full_3d/02_synapse_nnunet.ipynb index b16d1ee7..4d1e29bc 100644 --- a/notebooks/06_end_to_end/full_3d/02_synapse_nnunet.ipynb +++ b/notebooks/06_end_to_end/full_3d/02_synapse_nnunet.ipynb @@ -187,9 +187,6 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", - "\n", - "os.environ['MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR'] = 'false'\n", "\n", "from datamint.lightning import NNUNetTrainer\n", "\n", diff --git a/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb b/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb index 8333553e..81845dec 100644 --- a/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb +++ b/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb @@ -158,12 +158,9 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", "\n", "from datamint.lightning import EfficientNetV2Trainer\n", "\n", - "os.environ['MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR'] = 'false'\n", - "\n", "trainer = EfficientNetV2Trainer(\n", " project=PROJECT_NAME,\n", " image_size=384,\n", diff --git a/notebooks/06_end_to_end/slice_based/02_busi_segmentation.ipynb b/notebooks/06_end_to_end/slice_based/02_busi_segmentation.ipynb index b12ca423..b48ad515 100644 --- a/notebooks/06_end_to_end/slice_based/02_busi_segmentation.ipynb +++ b/notebooks/06_end_to_end/slice_based/02_busi_segmentation.ipynb @@ -187,12 +187,8 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", - "\n", "from datamint.lightning import DeepLabV3PlusTrainer\n", "\n", - "os.environ['MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR'] = 'false'\n", - "\n", "trainer = DeepLabV3PlusTrainer(\n", " project=PROJECT_NAME,\n", " image_size=256,\n", @@ -218,12 +214,8 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", - "\n", "from datamint.lightning import UNetPPTrainer\n", "\n", - "os.environ['MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR'] = 'false'\n", - "\n", "trainer = UNetPPTrainer(\n", " project=PROJECT_NAME,\n", " image_size=256,\n", @@ -249,12 +241,8 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", - "\n", "from datamint.lightning import TransUNetTrainer\n", "\n", - "os.environ['MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR'] = 'false'\n", - "\n", "trainer = TransUNetTrainer(\n", " project=PROJECT_NAME,\n", " # image_size is omitted \u2014 TransUNetTrainer always uses 224\u00d7224.\n", diff --git a/notebooks/06_end_to_end/slice_based/03_bccd_detection.ipynb b/notebooks/06_end_to_end/slice_based/03_bccd_detection.ipynb index c3994406..da999cff 100644 --- a/notebooks/06_end_to_end/slice_based/03_bccd_detection.ipynb +++ b/notebooks/06_end_to_end/slice_based/03_bccd_detection.ipynb @@ -150,12 +150,9 @@ "metadata": {}, "outputs": [], "source": [ - "import os\n", "\n", "from datamint.lightning import YOLOXTrainer\n", "\n", - "os.environ[\"MLFLOW_ENABLE_ARTIFACTS_PROGRESS_BAR\"] = \"false\"\n", - "\n", "trainer = YOLOXTrainer(\n", " project=PROJECT_NAME,\n", " model_size=\"s\", # nano / tiny / s / m / l / x\n",