Refactor on DoMINO utils.py to improve code quality

examples/cfd/external_aerodynamics/domino/requirements.txt

@@ -1,2 +1,4 @@
 torchinfo
-warp-lang
+warp-lang
+tensorboard
+cuml-cu12>=25.6.0

examples/cfd/external_aerodynamics/domino/src/inference_on_stl.py

@@ -34,11 +34,17 @@
 import numpy as np
 import torch
 
-from modulus.models.domino.model import DoMINO
-from modulus.utils.domino.utils import *
+from physicsnemo.models.domino.model import DoMINO
+from physicsnemo.utils.domino.utils import (
+    unnormalize,
+    create_directory,
+    nd_interpolator,
+    get_filenames,
+    write_to_vtp,
+)
 from torch.cuda.amp import autocast
 from torch.nn.parallel import DistributedDataParallel
-from modulus.distributed import DistributedManager
+from physicsnemo.distributed import DistributedManager
 
 from numpy.typing import NDArray
 from typing import Any, Iterable, List, Literal, Mapping, Optional, Union, Callable
@@ -49,7 +55,7 @@
 import pyvista as pv
 
 try:
-    from modulus.sym.geometry.tessellation import Tessellation
+    from physicsnemo.sym.geometry.tessellation import Tessellation
 
     SYM_AVAILABLE = True
 except ImportError:
@@ -663,7 +669,7 @@ def __init__(
         self,
         cfg: DictConfig,
         dist: None,
-        cached_geo_encoding: False,
+        cached_geo_encoding: bool = False,
     ):
 
         self.cfg = cfg

examples/cfd/external_aerodynamics/domino/src/train.py

@@ -265,7 +265,7 @@ def compute_loss_dict(
     integral_scaling_factor: float,
     surf_loss_scaling: float,
     vol_loss_scaling: float,
-) -> Tuple[torch.Tensor, dict]:
+) -> tuple[torch.Tensor, dict]:
     """
     Compute the loss terms in a single function call.
 

physicsnemo/datapipes/cae/domino_datapipe.py

@@ -843,7 +843,7 @@ def preprocess_volume(
                 mesh_indices_flattened,
                 grid_reshaped,
                 use_sign_winding_number=True,
-            ).reshape(nx, ny, nz)
+            ).reshape((nx, ny, nz))
 
             if self.config.sampling:
                 volume_coordinates_sampled, idx_volume = shuffle_array(