snowflake.ml.model.model_signature.DataType¶

class snowflake.ml.model.model_signature.DataType(value)¶

Bases: Enum

An enumeration.

Methods

as_snowpark_type() → DataType¶

Convert to corresponding Snowpark Type.

Returns:

A Snowpark type.

classmethod from_numpy_type(input_type: Union[dtype[Any], None, type[Any], _SupportsDType[dtype[Any]], str, tuple[Any, int], tuple[Any, Union[SupportsIndex, collections.abc.Sequence[SupportsIndex]]], list[Any], _DTypeDict, tuple[Any, Any], Int8Dtype, Int16Dtype, Int32Dtype, Int64Dtype, UInt8Dtype, UInt16Dtype, UInt32Dtype, UInt64Dtype, Float32Dtype, Float64Dtype, BooleanDtype, StringDtype]) → DataType¶

Translate numpy dtype to DataType for signature definition.

Parameters:

input_type – The numpy dtype or Pandas Extension Dtype

Raises:

SnowflakeMLException – NotImplementedError: Raised when the given numpy type is not supported.

Returns:

Corresponding DataType.

classmethod from_torch_type(torch_type: torch.dtype) → DataType¶
classmethod from_snowpark_type(snowpark_type: DataType) → DataType¶

Translate snowpark type to DataType for signature definition.

Parameters:

snowpark_type – The snowpark type.

Raises:

SnowflakeMLException – NotImplementedError: Raised when the given numpy type is not supported.

Returns:

Corresponding DataType.

Attributes

INT8 = ('int8', <class 'snowflake.snowpark.types.ByteType'>, <class 'numpy.int8'>)¶
INT16 = ('int16', <class 'snowflake.snowpark.types.ShortType'>, <class 'numpy.int16'>)¶
INT32 = ('int32', <class 'snowflake.snowpark.types.IntegerType'>, <class 'numpy.int32'>)¶
INT64 = ('int64', <class 'snowflake.snowpark.types.LongType'>, <class 'numpy.int64'>)¶
FLOAT = ('float', <class 'snowflake.snowpark.types.FloatType'>, <class 'numpy.float32'>)¶
DOUBLE = ('double', <class 'snowflake.snowpark.types.DoubleType'>, <class 'numpy.float64'>)¶
UINT8 = ('uint8', <class 'snowflake.snowpark.types.ByteType'>, <class 'numpy.uint8'>)¶
UINT16 = ('uint16', <class 'snowflake.snowpark.types.ShortType'>, <class 'numpy.uint16'>)¶
UINT32 = ('uint32', <class 'snowflake.snowpark.types.IntegerType'>, <class 'numpy.uint32'>)¶
UINT64 = ('uint64', <class 'snowflake.snowpark.types.LongType'>, <class 'numpy.uint64'>)¶
BOOL = ('bool', <class 'snowflake.snowpark.types.BooleanType'>, <class 'numpy.bool_'>)¶
STRING = ('string', <class 'snowflake.snowpark.types.StringType'>, <class 'numpy.str_'>)¶
BYTES = ('bytes', <class 'snowflake.snowpark.types.BinaryType'>, <class 'numpy.bytes_'>)¶
TIMESTAMP_NTZ = ('datetime64[ns]', <class 'snowflake.snowpark.types.TimestampType'>, 'datetime64[ns]')¶