From a9b44fdb78f6b79a9e52529d0b817cf3d987b8eb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pablo=20G=C3=B3mez?= Date: Mon, 25 Nov 2024 16:40:09 +0100 Subject: [PATCH] flaking --- torchquad/utils/set_precision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchquad/utils/set_precision.py b/torchquad/utils/set_precision.py index 68959e2c..b551aee6 100644 --- a/torchquad/utils/set_precision.py +++ b/torchquad/utils/set_precision.py @@ -68,7 +68,7 @@ def set_precision(data_type="float32", backend="torch"): logger.info(f"TensorFlow default floatx set to {tf.keras.backend.floatx()}") elif backend == "numpy": # NumPy still lacks global dtype support - os.environ[f"TORCHQUAD_DTYPE_NUMPY"] = data_type + os.environ["TORCHQUAD_DTYPE_NUMPY"] = data_type logger.info(f"NumPy default dtype set to {_get_precision('numpy')}") else: logger.error(f"Changing the data type is not supported for backend {backend}")