diff --git a/datamint/mlflow/__init__.py b/datamint/mlflow/__init__.py
index 3989739..94ddc21 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 f408f47..a6a51fa 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 b16d1ee..4d1e29b 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 8333553..81845de 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 b12ca42..b48ad51 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 c399440..da999cf 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",