-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Description
Describe the feature you'd like
We want sagemaker.workflow.parameters.ParameterInteger
to be able to get enum value like sagemaker.workflow.parameters.ParameterString
How would this feature be used? Please describe.
When using sagemaker.estimator.Estimator
, you need to pass a parameter named container_log_level based on integer values from https://docs.python.org/3/library/logging.html#levels . We extracted this parameter to a PipelineVariable. However, because it is a sagemaker.workflow.parameters.ParameterInteger
we cannot enforce the user to only select valid values are defined in the Python logging module. Enabling sagemaker.workflow.parameters.ParameterInteger
to have enum values will fix that.
Describe alternatives you've considered
Throwing exception in code we wrote for the estimator.