Skip to content

Commit a0d8d5b

Browse files
committed
Fix warning msg
1 parent 958661c commit a0d8d5b

File tree

1 file changed

+9
-12
lines changed

1 file changed

+9
-12
lines changed

bayesflow/__init__.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,16 @@
55
def setup():
66
# perform any necessary setup without polluting the namespace
77
import os
8+
import logging
89
from importlib.util import find_spec
910

11+
# set the basic logging level if the user hasn't already
12+
logging.basicConfig(level=logging.INFO)
13+
14+
# use a separate logger for the bayesflow package
15+
logger = logging.getLogger(__name__)
16+
logger.setLevel(logging.INFO)
17+
1018
issue_url = "https://github.com/bayesflow-org/bayesflow/issues/new?template=bug_report.md"
1119

1220
if "KERAS_BACKEND" not in os.environ:
@@ -45,13 +53,11 @@ def __init__(self, display_name, package_name, env_name, install_url, priority):
4553
raise ImportError(message)
4654

4755
if len(found_backends) > 1:
48-
import warnings
49-
5056
found_backends.sort(key=lambda b: b.priority)
5157
chosen_backend = found_backends[0]
5258
os.environ["KERAS_BACKEND"] = chosen_backend.env_name
5359

54-
warnings.warn(
60+
logging.warning(
5561
f"Multiple Keras-compatible backends detected ({', '.join(b.display_name for b in found_backends)}).\n"
5662
f"Defaulting to {chosen_backend.display_name}.\n"
5763
"To override, set the KERAS_BACKEND environment variable before importing bayesflow.\n"
@@ -61,15 +67,6 @@ def __init__(self, display_name, package_name, env_name, install_url, priority):
6167
os.environ["KERAS_BACKEND"] = found_backends[0].env_name
6268

6369
import keras
64-
import logging
65-
66-
# set the basic logging level if the user hasn't already
67-
logging.basicConfig(level=logging.INFO)
68-
69-
# use a separate logger for the bayesflow package
70-
logger = logging.getLogger(__name__)
71-
logger.setLevel(logging.INFO)
72-
7370
from bayesflow.utils import logging
7471

7572
if keras.backend.backend().lower() != os.environ["KERAS_BACKEND"].lower():

0 commit comments

Comments
 (0)