def to_flyte_idl(self) -> _idl_parameter_ranges.ParameterRanges: """ :rtype: _idl_parameter_ranges.ParameterRanges """ converted = {} for k, v in self._parameter_range_map.items(): if isinstance(v, IntegerParameterRange): converted[k] = _idl_parameter_ranges.ParameterRangeOneOf( integer_parameter_range=v.to_flyte_idl()) elif isinstance(v, ContinuousParameterRange): converted[k] = _idl_parameter_ranges.ParameterRangeOneOf( continuous_parameter_range=v.to_flyte_idl()) elif isinstance(v, CategoricalParameterRange): converted[k] = _idl_parameter_ranges.ParameterRangeOneOf( categorical_parameter_range=v.to_flyte_idl()) else: raise user.FlyteTypeException( received_type=type(v), expected_type=type(Union[IntegerParameterRange, ContinuousParameterRange, CategoricalParameterRange]), ) return _idl_parameter_ranges.ParameterRanges( parameter_range_map=converted, )
def to_flyte_idl(self) -> _idl_parameter_ranges.ParameterRanges: converted = {} for k, v in self._parameter_range_map.items(): if isinstance(v, IntegerParameterRange): converted[k] = _idl_parameter_ranges.ParameterRangeOneOf( integer_parameter_range=v.to_flyte_idl()) elif isinstance(v, ContinuousParameterRange): converted[k] = _idl_parameter_ranges.ParameterRangeOneOf( continuous_parameter_range=v.to_flyte_idl()) else: converted[k] = _idl_parameter_ranges.ParameterRangeOneOf( categorical_parameter_range=v.to_flyte_idl()) return _idl_parameter_ranges.ParameterRanges( parameter_range_map=converted, )
def to_flyte_idl(self) -> _idl_parameter_ranges.ParameterRangeOneOf: return _idl_parameter_ranges.ParameterRangeOneOf( integer_parameter_range=self.integer_parameter_range.to_flyte_idl( ) if self.integer_parameter_range else None, continuous_parameter_range=self.continuous_parameter_range. to_flyte_idl() if self.continuous_parameter_range else None, categorical_parameter_range=self.categorical_parameter_range. to_flyte_idl() if self.categorical_parameter_range else None, )
def to_python_value( self, ctx: FlyteContext, lv: Literal, expected_python_type: Type[_params.ParameterRangeOneOf] ) -> _params.ParameterRangeOneOf: if lv and lv.scalar and lv.scalar.generic is not None: d = json.loads(json_format.MessageToJson(lv.scalar.generic)) o = _pb2_params.ParameterRangeOneOf() o = json_format.ParseDict(d, o) return _params.ParameterRangeOneOf.from_flyte_idl(o) return None