Project import generated by Copybara. (#97)

GitOrigin-RevId: d335dcb33796fa8ad29bfde02530d08bc3c4b92e

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

Author: sfc-gh-anavalos

CHANGELOG.md

@@ -1,6 +1,7 @@
 workspace(name = "SnowML")
 
 load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive", "http_jar")
+load("@bazel_tools//tools/build_defs/repo:git.bzl", "git_repository")
 
 http_jar(
     name = "bazel_diff",
@@ -51,6 +52,23 @@ aspect_bazel_lib_dependencies()
 
 register_yq_toolchains()
 
+git_repository(
+    name = "bazel_latex",
+    commit = "f9e1532513760f1379396e55e9e38013869d2a8c",
+    remote = "https://github.com/ProdriveTechnologies/bazel-latex.git",
+)
+
+load("@bazel_latex//:repositories.bzl", "latex_repositories")
+
+latex_repositories()
+
+register_toolchains(
+    "@bazel_latex//:latex_toolchain_aarch64-darwin",
+    "@bazel_latex//:latex_toolchain_amd64-freebsd",
+    "@bazel_latex//:latex_toolchain_x86_64-darwin",
+    "@bazel_latex//:latex_toolchain_x86_64-linux",
+)
+
 local_repository(
     name = "rules_sphinx",
     path = "./third_party/rules_sphinx",

WORKSPACE

@@ -12,6 +12,7 @@ dependencies:
   - boto3==1.24.28
   - build==0.10.0
   - cachetools==4.2.2
+  - catboost==1.2.0
   - cloudpickle==2.2.1
   - conda-libmamba-solver==23.7.0
   - coverage==6.3.2

bazel/environments/conda-env-snowflake.yml

@@ -12,6 +12,7 @@ dependencies:
   - boto3==1.24.28
   - build==0.10.0
   - cachetools==4.2.2
+  - catboost==1.2.0
   - cloudpickle==2.2.1
   - conda-forge::accelerate==0.22.0
   - conda-forge::mypy==1.5.1

bazel/environments/conda-env.yml

@@ -12,6 +12,7 @@ dependencies:
   - boto3==1.24.28
   - build==0.10.0
   - cachetools==4.2.2
+  - catboost==1.2.0
   - cloudpickle==2.2.1
   - conda-forge::accelerate==0.22.0
   - conda-forge::mypy==1.5.1

bazel/environments/conda-gpu-env.yml

@@ -95,8 +95,7 @@
             {
               "enum": [
                 "deployment_core",
-                "udf_inference",
-                "spcs_inference",
+                "snowml_inference_alternative",
                 "model_packaging",
                 "build_essential",
                 "build_test_env"

bazel/requirements/requirements.schema.json

@@ -14,8 +14,6 @@ requirements:
     - bazel >=6.0.0
   run:
     - python>=3.8,<3.12
-  run_constrained:
-    - pytorch<2.1.0  # [win]
 
 about:
   home: https://github.com/snowflakedb/snowflake-ml-python

bazel/requirements/templates/meta.tpl.yaml

@@ -17,7 +17,7 @@ build:
   noarch: python
 package:
   name: snowflake-ml-python
-  version: 1.4.0
+  version: 1.4.1
 requirements:
   build:
     - python
@@ -27,12 +27,13 @@ requirements:
     - aiohttp!=4.0.0a0, !=4.0.0a1
     - anyio>=3.5.0,<4
     - cachetools>=3.1.1,<6
+    - catboost>=1.2.0, <1.3
     - cloudpickle>=2.0.0
     - fsspec>=2022.11,<2024
     - importlib_resources>=6.1.1, <7
     - numpy>=1.23,<2
     - packaging>=20.9,<24
-    - pandas>=1.0.0,<2
+    - pandas>=1.0.0,<3
     - pyarrow
     - pytimeparse>=1.1.8,<2
     - pyyaml>=6.0,<7
@@ -48,15 +49,15 @@ requirements:
     - xgboost>=1.7.3,<2
     - python>=3.8,<3.12
   run_constrained:
-    - lightgbm==3.3.5
+    - lightgbm>=3.3.5,<4.2
     - mlflow>=2.1.0,<2.4
+    - pytorch>=2.0.1,<3
     - sentence-transformers>=2.2.2,<3
     - sentencepiece>=0.1.95,<0.2
     - shap==0.42.1
     - tensorflow>=2.10,<3
     - tokenizers>=0.10,<1
     - torchdata>=0.4,<1
     - transformers>=4.32.1,<5
-    - pytorch<2.1.0 # [win]
 source:
   path: ../../