🐛[BUG]: Unable to use `torch.where` expression when constructing PDEs. Workaround is to use Heaviside.

[AAKHASH17]

Version

2.3.0

On which installation method(s) does this occur?

Pip

Describe the issue

Hi Nvidia - Physicsnemo-Sym Developers,
I'm unable to use the Piecewise sympy expression to invoke the torch.where functionality when constructing the PDEs. As an alternative I found that using the Heaviside works but the usage of torch.where is required when having conditional evaluation of tensor operations. Can someone please help me with this?

Thanks

Minimum reproducible example

Relevant log output

Environment details

Other/Misc.

No response

[ram-cherukuri]

@AAKHASH17 Can you share more details like a minimum reproducible example to help address this?

[ktangsali]

Closing as part of the archival of this repository. The key components have been upstreamed to NVIDIA/physicsnemo. Please refer to the v2.0 Migration Guide and feel free to re-open your issue there if it is still relevant.