From cae7c0ae5082eab85607e62cb01af60f2eeb5b61 Mon Sep 17 00:00:00 2001 From: wenkelf Date: Thu, 12 Oct 2023 04:32:39 +0000 Subject: [PATCH] Various updates --- graphium/nn/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphium/nn/utils.py b/graphium/nn/utils.py index 68a8779c4..e9ac4fa0c 100644 --- a/graphium/nn/utils.py +++ b/graphium/nn/utils.py @@ -40,7 +40,7 @@ def scale_kwargs(self, scale_factor: Real, scale_in_dim: bool = False): divide_factor = 1 / scale_factor - if scale_in_dim is None: + if not scale_in_dim: return self.make_mup_base_kwargs(divide_factor=divide_factor) # If scale_in_dim passed, need to check it can be forwarded