Project import generated by Copybara. (#83)

GitOrigin-RevId: c4f12870743148f784972f89d486a79c3cf4d5b9

Co-authored-by: Snowflake Authors <[email protected]>

.markdownlint.yaml

@@ -0,0 +1,243 @@
+---
+# Example markdownlint configuration with all properties set to their default value
+
+# Default state for all rules
+default: true
+
+# Path to configuration file to extend
+extends:
+# MD001/heading-increment/header-increment - Heading levels should only increment by one level at a time
+MD001: true
+
+# MD002/first-heading-h1/first-header-h1 - First heading should be a top-level heading
+MD002:
+  # Heading level
+  level: 1
+
+# MD003/heading-style/header-style - Heading style
+MD003:
+  # Heading style
+  style: consistent
+
+# MD004/ul-style - Unordered list style
+MD004:
+  # List style
+  style: consistent
+# MD005/list-indent - Inconsistent indentation for list items at the same level
+MD005: true
+
+# MD006/ul-start-left - Consider starting bulleted lists at the beginning of the line
+MD006: true
+
+# MD007/ul-indent - Unordered list indentation
+MD007:
+  # Spaces for indent
+  indent: 2
+  # Whether to indent the first level of the list
+  start_indented: false
+  # Spaces for first level indent (when start_indented is set)
+  start_indent: 2
+
+# MD009/no-trailing-spaces - Trailing spaces
+MD009:
+  # Spaces for line break
+  br_spaces: 2
+  # Allow spaces for empty lines in list items
+  list_item_empty_lines: false
+  # Include unnecessary breaks
+  strict: false
+
+# MD010/no-hard-tabs - Hard tabs
+MD010:
+  # Include code blocks
+  code_blocks: true
+  # Fenced code languages to ignore
+  ignore_code_languages: []
+  # Number of spaces for each hard tab
+  spaces_per_tab: 2
+# MD011/no-reversed-links - Reversed link syntax
+MD011: true
+
+# MD012/no-multiple-blanks - Multiple consecutive blank lines
+MD012:
+  # Consecutive blank lines
+  maximum: 1
+
+# MD013/line-length - Line length
+MD013:
+  # Number of characters
+  line_length: 120
+  # Number of characters for headings
+  heading_line_length: 80
+  # Number of characters for code blocks
+  code_block_line_length: 120
+  # Include code blocks
+  code_blocks: true
+  # Include tables
+  tables: true
+  # Include headings
+  headings: true
+  # Include headings
+  headers: true
+  # Strict length checking
+  strict: false
+  # Stern length checking
+  stern: false
+# MD014/commands-show-output - Dollar signs used before commands without showing output
+MD014: true
+
+# MD018/no-missing-space-atx - No space after hash on atx style heading
+MD018: true
+
+# MD019/no-multiple-space-atx - Multiple spaces after hash on atx style heading
+MD019: true
+
+# MD020/no-missing-space-closed-atx - No space inside hashes on closed atx style heading
+MD020: true
+
+# MD021/no-multiple-space-closed-atx - Multiple spaces inside hashes on closed atx style heading
+MD021: true
+
+# MD022/blanks-around-headings/blanks-around-headers - Headings should be surrounded by blank lines
+MD022:
+  # Blank lines above heading
+  lines_above: 1
+  # Blank lines below heading
+  lines_below: 1
+# MD023/heading-start-left/header-start-left - Headings must start at the beginning of the line
+MD023: true
+
+# MD024/no-duplicate-heading/no-duplicate-header - Multiple headings with the same content
+MD024:
+  # Only check sibling headings
+  allow_different_nesting: true
+  # Only check sibling headings
+  siblings_only: false
+
+# MD025/single-title/single-h1 - Multiple top-level headings in the same document
+MD025:
+  # Heading level
+  level: 1
+  # RegExp for matching title in front matter
+  front_matter_title: ^\s*title\s*[:=]
+
+# MD026/no-trailing-punctuation - Trailing punctuation in heading
+MD026:
+  # Punctuation characters
+  punctuation: .,;:!。，；：！
+# MD027/no-multiple-space-blockquote - Multiple spaces after blockquote symbol
+MD027: true
+
+# MD028/no-blanks-blockquote - Blank line inside blockquote
+MD028: true
+
+# MD029/ol-prefix - Ordered list item prefix
+MD029:
+  # List style
+  style: one_or_ordered
+
+# MD030/list-marker-space - Spaces after list markers
+MD030:
+  # Spaces for single-line unordered list items
+  ul_single: 1
+  # Spaces for single-line ordered list items
+  ol_single: 1
+  # Spaces for multi-line unordered list items
+  ul_multi: 1
+  # Spaces for multi-line ordered list items
+  ol_multi: 1
+
+# MD031/blanks-around-fences - Fenced code blocks should be surrounded by blank lines
+MD031:
+  # Include list items
+  list_items: true
+# MD032/blanks-around-lists - Lists should be surrounded by blank lines
+MD032: true
+
+# MD033/no-inline-html - Inline HTML
+MD033:
+  # Allowed elements
+  allowed_elements: []
+# MD034/no-bare-urls - Bare URL used
+MD034: true
+
+# MD035/hr-style - Horizontal rule style
+MD035:
+  # Horizontal rule style
+  style: consistent
+
+# MD036/no-emphasis-as-heading/no-emphasis-as-header - Emphasis used instead of a heading
+MD036:
+  # Punctuation characters
+  punctuation: .,;:!?。，；：！？
+# MD037/no-space-in-emphasis - Spaces inside emphasis markers
+MD037: true
+
+# MD038/no-space-in-code - Spaces inside code span elements
+MD038: true
+
+# MD039/no-space-in-links - Spaces inside link text
+MD039: true
+
+# MD040/fenced-code-language - Fenced code blocks should have a language specified
+MD040:
+  # List of languages
+  allowed_languages: []
+  # Require language only
+  language_only: false
+
+# MD041/first-line-heading/first-line-h1 - First line in a file should be a top-level heading
+MD041:
+  # Heading level
+  level: 1
+  # RegExp for matching title in front matter
+  front_matter_title: ^\s*title\s*[:=]
+# MD042/no-empty-links - No empty links
+MD042: true
+
+# MD043/required-headings/required-headers - Required heading structure
+MD043: false
+
+# MD044/proper-names - Proper names should have the correct capitalization
+MD044:
+  # List of proper names
+  names: []
+  # Include code blocks
+  code_blocks: true
+  # Include HTML elements
+  html_elements: true
+# MD045/no-alt-text - Images should have alternate text (alt text)
+MD045: true
+
+# MD046/code-block-style - Code block style
+MD046:
+  # Block style
+  style: consistent
+# MD047/single-trailing-newline - Files should end with a single newline character
+MD047: true
+
+# MD048/code-fence-style - Code fence style
+MD048:
+  # Code fence style
+  style: consistent
+
+# MD049/emphasis-style - Emphasis style should be consistent
+MD049:
+  # Emphasis style
+  style: consistent
+
+# MD050/strong-style - Strong style should be consistent
+MD050:
+  # Strong style
+  style: consistent
+# MD051/link-fragments - Link fragments should be valid
+MD051: true
+
+# MD052/reference-links-images - Reference links and images should use a label that is defined
+MD052: true
+
+# MD053/link-image-reference-definitions - Link and image reference definitions should be needed
+MD053:
+  # Ignored definitions
+  ignored_definitions:
+    - //

BUILD.bazel

@@ -1,8 +1,12 @@
 load("@rules_java//java:defs.bzl", "java_binary")
+load("//:packages.bzl", "PACKAGES")
+load("//bazel:py_rules.bzl", "py_wheel")
+load("//bazel/requirements:rules.bzl", "generate_pyproject_file")
 
 exports_files([
     "CHANGELOG.md",
     "README.md",
+    "LICENSE.txt",
     "conda-env-extended.yml",
     "conda-env-snowflake.yml",
     "conda-env.yml",
@@ -16,3 +20,19 @@ java_binary(
     main_class = "com.bazel_diff.Main",
     runtime_deps = ["@bazel_diff//jar"],
 )
+
+generate_pyproject_file(
+    name = "snowml",
+    src_requirement_file = "//:requirements.yml",
+)
+
+py_wheel(
+    name = "wheel",
+    data = [
+        "//:CHANGELOG.md",
+        "//:LICENSE.txt",
+        "//:README.md",
+    ],
+    pyproject_toml = ":snowml_pyproject",
+    deps = PACKAGES,
+)

CHANGELOG.md

@@ -1,5 +1,12 @@
 # Release History
 
+## 1.2.1
+
+### New Features
+
+- Model Development: Infers output column data type for transformers when possible.
+- Registry: `relax_version` option is available in the `options` argument when logging the model.
+
 ## 1.2.0
 
 ### Bug Fixes

README.md

@@ -10,7 +10,7 @@ Learning workflow.
 The Snowpark ML Python SDK provides a number of APIs to support each stage of an end-to-end Machine Learning development
 and deployment process, and includes two key components.
 
-### Snowpark ML Development [Public Preview]
+### Snowpark ML Development
 
 [Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-development)
 provides a collection of python APIs enabling efficient ML model development directly in Snowflake:

_typos.toml

@@ -0,0 +1,16 @@
+[default]
+extend-ignore-identifiers-re = [
+  ".*[A-Fa-f0-9]{8,}"
+]
+
+[default.extend-identifiers]
+LiteralNDArrayType = "LiteralNDArrayType"
+
+[default.extend-words]
+MAPE = "MAPE"
+fpr = "fpr"
+fwe = "fwe"
+
+[type.po]
+extend-glob = ["*.ipynb"]
+check-file = false

bazel/BUILD.bazel

@@ -2,6 +2,10 @@ load("@rules_python//python:defs.bzl", native_py_test = "py_test")
 
 package(default_visibility = ["//visibility:public"])
 
+exports_files([
+    "wheelbuilder.py",
+])
+
 native_py_test(
     name = "repo_paths_test",
     srcs = ["repo_paths_test.py"],
@@ -19,8 +23,9 @@ sh_binary(
 package_group(
     name = "snowml_public_common",
     packages = [
-        "//bazel/...",
-        "//ci/...",
-        "//docs/...",
+        "-//codegen/...",
+        "-//snowflake/...",
+        "-//tests/...",
+        "//...",
     ],
 )

bazel/environments/conda-env-build.yml

@@ -7,6 +7,7 @@ channels:
   - nodefaults
 dependencies:
   - absl-py==1.3.0
+  - build==0.10.0
   - conda-libmamba-solver==23.7.0
   - inflection==0.5.1
   - jsonschema==3.2.0
@@ -16,4 +17,6 @@ dependencies:
   - ruamel.yaml==0.17.21
   - scikit-learn==1.3.0
   - sphinx==5.0.2
+  - toml==0.10.2
+  - types-toml==0.10.8.6
   - xgboost==1.7.3

bazel/environments/conda-env-snowflake.yml

@@ -10,6 +10,7 @@ dependencies:
   - aiohttp==3.8.3
   - anyio==3.5.0
   - boto3==1.24.28
+  - build==0.10.0
   - cachetools==4.2.2
   - cloudpickle==2.0.0
   - conda-libmamba-solver==23.7.0
@@ -50,9 +51,11 @@ dependencies:
   - sqlparse==0.4.4
   - tensorflow==2.10.0
   - tokenizers==0.13.2
+  - toml==0.10.2
   - torchdata==0.6.1
   - transformers==4.32.1
   - types-protobuf==4.23.0.1
   - types-requests==2.30.0.0
+  - types-toml==0.10.8.6
   - typing-extensions==4.5.0
   - xgboost==1.7.3

bazel/environments/conda-env.yml

@@ -10,6 +10,7 @@ dependencies:
   - aiohttp==3.8.3
   - anyio==3.5.0
   - boto3==1.24.28
+  - build==0.10.0
   - cachetools==4.2.2
   - cloudpickle==2.0.0
   - conda-forge::accelerate==0.22.0
@@ -55,10 +56,12 @@ dependencies:
   - sqlparse==0.4.4
   - tensorflow==2.10.0
   - tokenizers==0.13.2
+  - toml==0.10.2
   - torchdata==0.6.1
   - transformers==4.32.1
   - types-protobuf==4.23.0.1
   - types-requests==2.30.0.0
+  - types-toml==0.10.8.6
   - typing-extensions==4.5.0
   - xgboost==1.7.3
   - pip