fix optional postprocess

openforcefield · Nov 4, 2024 · 0e16ffc · 0e16ffc
1 parent 6e4bad6
commit 0e16ffc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/openff/nagl/nn/_containers.py b/openff/nagl/nn/_containers.py
@@ -179,6 +179,7 @@ def from_config(
         layer_dropout = [
             layer.dropout for layer in readout_config.layers
         ]
+        postprocess_layer = None
         if readout_config.postprocess is not None:
             postprocess_layer = _PostprocessLayerMeta._get_object(readout_config.postprocess)
             hidden_feature_sizes.append(postprocess_layer.n_features)