Domino-training-test errors while executing

[FlowTwinAI]

Highlighted blocks cause errors while executing

1. Data Preprocessing [domino-data-preprocessing.ipynb]
   -(old) print(f"Number of Faces: {mesh1.n_faces} vs {mesh2.n_faces}")
   -(new) print(f"Number of Faces: {mesh1.n_cells} vs {mesh2.n_cells}")

2. Training -Test [domino-training-test.ipynb]

[!ERROR]
/usr/local/lib/python3.12/dist-packages/physicsnemo/distributed/manager.py:392: UserWarning: Distributed manager is already intialized
warn("Distributed manager is already intialized")
AttributeError Traceback (most recent call last)
Cell In[5], line 16
14 print(device)
15 # Set up model
---> 16 model = create_model(device, rank, world_size)
17 # Run training
18 train(model, device, rank, world_size)
Cell In[3], line 94, in create_model(device, rank, world_size)
80 """
81 Create and configure DoMINO model with distributed training support.
82
(...) 89 DoMINO: Configured model (wrapped in DistributedDataParallel if world_size > 1)
90 """
93 # Initialize model with configuration
---> 94 model = DoMINO(
95 input_features=3,
96 output_features_vol=None,
97 output_features_surf=NUM_SURF_VARS,
98 model_parameters=SimpleNamespace(
99 interp_res=GRID_RESOLUTION,
100 surface_neighbors=NUM_SURFACE_NEIGHBORS,
101 use_surface_normals=True,
102 use_surface_area=True,
103 encode_parameters=True,
104 positional_encoding=False,
105 integral_loss_scaling_factor=INTEGRAL_LOSS_SCALING,
106 normalization=NORMALIZATION,
107 use_sdf_in_basis_func=True,
108 geometry_encoding_type= GEOMETRY_ENCODING_TYPE, # geometry encoder type, sdf, stl, both
109 geometry_rep=GEOMETRY_REP,
110 nn_basis_functions=SimpleNamespace(base_layer=512, fourier_features=False, num_modes=5),
111 parameter_model=SimpleNamespace(base_layer=512, scaling_params=[30.0, 1.226], fourier_features=False, num_modes=5),
112 position_encoder=SimpleNamespace(base_neurons=512),
113 geometry_local=GEOMETRY_LOCAL,
114 aggregation_model=SimpleNamespace(base_layer=512),
115 model_type=MODEL_TYPE
116 ),
117 ).to(device)
119 # Wrap model for distributed training if needed
120 if world_size > 1:
File /usr/local/lib/python3.12/dist-packages/physicsnemo/models/domino/model.py:1016, in DoMINO.init(self, input_features, output_features_vol, output_features_surf, global_features, model_parameters)
1014 self.output_features_vol = output_features_vol
1015 self.output_features_surf = output_features_surf
-> 1016 self.num_sample_points_surface = model_parameters.num_neighbors_surface
1017 self.num_sample_points_volume = model_parameters.num_neighbors_volume
1018 self.combined_vol_surf = model_parameters.combine_volume_surface
AttributeError: 'types.SimpleNamespace' object has no attribute 'num_neighbors_surface'
/