From e297fad279423b63a82f9efabca247253a301252 Mon Sep 17 00:00:00 2001 From: frostedoyster Date: Thu, 29 Aug 2024 12:52:13 +0200 Subject: [PATCH] `._module` -> `.module` --- src/metatrain/experimental/pet/tests/test_pet_compatibility.py | 2 +- src/metatrain/utils/llpr.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/metatrain/experimental/pet/tests/test_pet_compatibility.py b/src/metatrain/experimental/pet/tests/test_pet_compatibility.py index 16f7955a9..852750be9 100644 --- a/src/metatrain/experimental/pet/tests/test_pet_compatibility.py +++ b/src/metatrain/experimental/pet/tests/test_pet_compatibility.py @@ -159,7 +159,7 @@ def test_predictions_compatibility(cutoff): "neighbors_pos": batch.neighbors_pos, } - pet = model._module.pet + pet = model.module.pet pet_prediction = pet.forward(batch_dict) diff --git a/src/metatrain/utils/llpr.py b/src/metatrain/utils/llpr.py index 20eb68df6..7f23fda9b 100644 --- a/src/metatrain/utils/llpr.py +++ b/src/metatrain/utils/llpr.py @@ -31,7 +31,7 @@ def __init__( super().__init__() self.model = model - self.ll_feat_size = self.model._module.last_layer_feature_size + self.ll_feat_size = self.model.module.last_layer_feature_size # update capabilities: now we have additional outputs for the uncertainty old_capabilities = self.model.capabilities()