diff --git a/moose/kernels.py b/moose/kernels.py index 7eb490f..7b5e4c3 100644 --- a/moose/kernels.py +++ b/moose/kernels.py @@ -158,7 +158,7 @@ def __str__(self): string = f'[{self.name}]\n' string += f'type={self.aux_kernel_type.name}\n' string += f'variable={self.variable.name}\n' - string += f'{self.function.__str__()}' + string += f'function={self.function.__str__()}\n' string += f'args={self.args.name}\n' string += '[]\n' return string diff --git a/moose/mesh.py b/moose/mesh.py index daaa5a2..b8a28ea 100644 --- a/moose/mesh.py +++ b/moose/mesh.py @@ -88,11 +88,14 @@ def __str__(self): def add_mesh_object(self, name = "", type = None, **kwargs): if name in self.mesh_objects.keys(): print (f'name {name} already in use') - + + mesh = None # Add a default value for the 'mesh' variable + if type == MeshObjectTypes.FileMeshGenerator: mesh = FileMeshGenerator(name=name, **kwargs) elif type == MeshObjectTypes.TransformGenerator: mesh = TransformGenerator(name=name, **kwargs) - self.mesh_objects[name] = mesh - \ No newline at end of file + if mesh is not None: + self.mesh_objects[name] = mesh +