🐛[BUG]: BallQueryLayer not found on the DoMiNo Fine Tuning Example

[phermosomore]

Version

main

On which installation method(s) does this occur?

Source

Describe the issue

BallQueryLayer not found when running https://github.com/NVIDIA/physicsnemo/blob/main/examples/cfd/external_aerodynamics/domino_nim_finetuning/src/generate_base_predictions.py

Minimum reproducible example

Relevant log output

Environment details