From 355e4fdeec5c43cf5c6770b09311818a9793c5d4 Mon Sep 17 00:00:00 2001 From: frostedoyster Date: Tue, 4 Jun 2024 11:46:01 +0200 Subject: [PATCH] Consistent capitalization --- src/metatensor/models/experimental/soap_bpnn/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/metatensor/models/experimental/soap_bpnn/trainer.py b/src/metatensor/models/experimental/soap_bpnn/trainer.py index 6ffb3004d2..041f6f5188 100644 --- a/src/metatensor/models/experimental/soap_bpnn/trainer.py +++ b/src/metatensor/models/experimental/soap_bpnn/trainer.py @@ -52,7 +52,7 @@ def train( assert len(devices) == 1 device = devices[0] - logger.info(f"training on device {device} with dtype {dtype}") + logger.info(f"Training on device {device} with dtype {dtype}") model.to(device=device, dtype=dtype) # Calculate and set the composition weights for all targets: