From 155936d3efdee94916dac05fc6d5aaf4815d2755 Mon Sep 17 00:00:00 2001 From: Leif Denby Date: Wed, 26 Feb 2025 09:54:53 +0100 Subject: [PATCH] ignore SSL cert auth when using mlflow logger --- neural_lam/utils.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/neural_lam/utils.py b/neural_lam/utils.py index f0b09ad38..8f459c41f 100644 --- a/neural_lam/utils.py +++ b/neural_lam/utils.py @@ -6,6 +6,7 @@ # Third-party import pytorch_lightning as pl import torch +import urllib3 from pytorch_lightning.loggers import MLFlowLogger, WandbLogger from pytorch_lightning.utilities import rank_zero_only from torch import nn @@ -297,6 +298,11 @@ def setup_training_logger(datastore, args, run_name): raise ValueError( "MLFlow logger requires setting MLFLOW_TRACKING_URI in env." ) + + # suppress warnings about insecure requests so that we avoid warnings in + # the logs when tracking on the MLflow tracking server + urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) + logger = CustomMLFlowLogger( experiment_name=args.logger_project, tracking_uri=url,