Project import generated by Copybara. (#15)

GitOrigin-RevId: 6655b0eee1ece39b1328ee13ba4ab1e09b8623a8 Co-authored-by: Snowflake Authors <[email protected]>
snowflakedb · May 19, 2023 · 2a14eaf · 2a14eaf
1 parent e9aac2d
commit 2a14eaf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ci/conda_recipe/meta.yaml b/ci/conda_recipe/meta.yaml
@@ -2,7 +2,7 @@
 # (dependencies, version number) from a common place. We also need to define that
 # common place, as currently it's a BUILD rule.
 # See https://docs.conda.io/projects/conda-build/en/latest/resources/define-metadata.html#templating-with-jinja
-{% set version_match = load_file_regex(load_file='../../snowflake/ml/version.bzl', regex_pattern='VERSION = "(\d\.\d\.\d*)"\s.*') %}
+{% set version_match = load_file_regex(load_file='snowflake/ml/version.bzl', regex_pattern='VERSION = "(\d\.\d\.\d*)"\s.*') %}
 
 package:
   name: snowflake-ml-python

diff --git a/snowflake/ml/model/model_signature.py b/snowflake/ml/model/model_signature.py
@@ -19,13 +19,15 @@ def __init__(self, value: str, sql_type: str, snowpark_type: spt.DataType, numpy
         self._snowpark_type = snowpark_type
         self._numpy_type = numpy_type
 
+    INT8 = ("int8", "INTEGER", spt.IntegerType(), np.int8)
     INT16 = ("int16", "INTEGER", spt.IntegerType(), np.int16)
     INT32 = ("int32", "INTEGER", spt.IntegerType(), np.int32)
     INT64 = ("int64", "INTEGER", spt.IntegerType(), np.int64)
 
     FLOAT = ("float", "FLOAT", spt.FloatType(), np.float32)
     DOUBLE = ("double", "DOUBLE", spt.DoubleType(), np.float64)
 
+    UINT8 = ("uint8", "INTEGER", spt.IntegerType(), np.uint8)
     UINT16 = ("uint16", "INTEGER", spt.IntegerType(), np.uint16)
     UINT32 = ("uint32", "INTEGER", spt.IntegerType(), np.uint32)
     UINT64 = ("uint64", "INTEGER", spt.IntegerType(), np.uint64)