ngen.cal Objective Enum needs type constraints

[aaraney]

ngen-cal/python/ngen_cal/strategy.py

Line 19 in fdca562

class Objective(Enum):

This enum should be a class Objective(str, Enum).

When Objective is used as a type in the Estimation pydantic model, the resulting json_schema does not qualify that that the objective field is constrained to string types.

  "definitions": {
    // ...
    "Objective": {
      "title": "Objective",
      "description": "Enumeration of supported search algorithms\n\n    ",
      // missing "type": "string",
      "enum": [
        "custom",
        "kling_gupta",
        "nnse",
        "single_peak",
        "volume"
      ]
    },
    "Estimation": {
      "title": "Estimation",
      "description": "Estimation strategy for defining parameter estimation",
      "type": "object",
      "properties": {
        // ...
        "objective": {
          "title": "Objective",
          "default": "custom",
          "anyOf": [
            {
              "$ref": "#/definitions/Objective"
            },
            // ...
          ]
        },
    // ...