🐛[BUG]:An activation function problem occurred when calling DGMArch in modulus-sym

[YouqiongLiu]

Version

modulus:0.6.0, modulus-sym:1.5.0

On which installation method(s) does this occur?

Pip

Describe the issue

RuntimeError:
function cannot be used as a value:
File "/home/envs/lib/python3.10/site-packages/modulus/models/layers/dgm_layers.py", line 90
x = self.linear_1(input_1) + self.linear_2(input_2) + self.bias

    if self.activation_fn is not None:
       ~~~~~~~~~~~~~~~~~~ <--- HERE
        if self.activation_par is None:
            x = self.activation_fn(x)

Minimum reproducible example

Relevant log output

Environment details

Other/Misc.

No response