diff --git a/src/ansys/fluent/core/codegen/datamodelgen.py b/src/ansys/fluent/core/codegen/datamodelgen.py index 532c2ef810b..781be261648 100644 --- a/src/ansys/fluent/core/codegen/datamodelgen.py +++ b/src/ansys/fluent/core/codegen/datamodelgen.py @@ -331,12 +331,12 @@ def _write_static_info(self, name: str, info: Any, f: FileIO, level: int = 0): # print("\t\texcluded", k) pass for k in parameters: - k_type = _PY_TYPE_BY_DM_TYPE[info["parameters"][k]["type"]] - if k_type in ["str", "list[str]"]: + k_type = info["parameters"][k]["type"] + if k_type in {"String", "String List", "ListString"}: f.write(f"{indent} class {k}(PyTextual):\n") - elif k_type in ["int", "float"]: + elif k_type in {"Integer", "Int", "Real"}: f.write(f"{indent} class {k}(PyNumerical):\n") - elif k_type in ["Dict", "dict[str, Any]"]: + elif k_type in {"Dict", "ModelObject"}: f.write(f"{indent} class {k}(PyDictionary):\n") else: f.write(f"{indent} class {k}(PyParameter):\n")