diff --git a/python/ray/air/util/data_batch_conversion.py b/python/ray/air/util/data_batch_conversion.py index 426814d72f1d..e134b5b1d31f 100644 --- a/python/ray/air/util/data_batch_conversion.py +++ b/python/ray/air/util/data_batch_conversion.py @@ -319,7 +319,7 @@ def _cast_ndarray_columns_to_tensor_extension(df: "pd.DataFrame") -> "pd.DataFra with warnings.catch_warnings(): warnings.simplefilter("ignore", category=FutureWarning) warnings.simplefilter("ignore", category=SettingWithCopyWarning) - df.loc[:, col_name] = TensorArray(col) + df[col_name] = TensorArray(col) except Exception as e: raise ValueError( f"Tried to cast column {col_name} to the TensorArray tensor " @@ -354,5 +354,5 @@ def _cast_tensor_columns_to_ndarrays(df: "pd.DataFrame") -> "pd.DataFrame": with warnings.catch_warnings(): warnings.simplefilter("ignore", category=FutureWarning) warnings.simplefilter("ignore", category=SettingWithCopyWarning) - df.loc[:, col_name] = pd.Series(list(col.to_numpy())) + df[col_name] = list(col.to_numpy()) return df diff --git a/python/ray/data/tests/test_tensor.py b/python/ray/data/tests/test_tensor.py index 852142bf4150..3932d5b76ac0 100644 --- a/python/ray/data/tests/test_tensor.py +++ b/python/ray/data/tests/test_tensor.py @@ -564,7 +564,7 @@ def test_tensors_in_tables_pandas_roundtrip( ds_df = ds.to_pandas() expected_df = df + 1 if enable_automatic_tensor_extension_cast: - expected_df.loc[:, "two"] = list(expected_df["two"].to_numpy()) + expected_df["two"] = list(expected_df["two"].to_numpy()) pd.testing.assert_frame_equal(ds_df, expected_df) @@ -585,7 +585,7 @@ def test_tensors_in_tables_pandas_roundtrip_variable_shaped( ds_df = ds.to_pandas() expected_df = df + 1 if enable_automatic_tensor_extension_cast: - expected_df.loc[:, "two"] = _create_possibly_ragged_ndarray( + expected_df["two"] = _create_possibly_ragged_ndarray( expected_df["two"].to_numpy() ) pd.testing.assert_frame_equal(ds_df, expected_df) @@ -873,8 +873,8 @@ def test_tensors_in_tables_iter_batches( ) df = pd.concat([df1, df2], ignore_index=True) if enable_automatic_tensor_extension_cast: - df.loc[:, "one"] = list(df["one"].to_numpy()) - df.loc[:, "two"] = list(df["two"].to_numpy()) + df["one"] = list(df["one"].to_numpy()) + df["two"] = list(df["two"].to_numpy()) ds = ray.data.from_pandas([df1, df2]) batches = list(ds.iter_batches(batch_size=2, batch_format="pandas")) assert len(batches) == 3 diff --git a/python/requirements_compiled.txt b/python/requirements_compiled.txt index c5cee09e4dc9..acc79216edb6 100644 --- a/python/requirements_compiled.txt +++ b/python/requirements_compiled.txt @@ -4,21 +4,40 @@ absl-py==1.4.0 # via # array-record + # chex + # dm-control + # dm-env + # dopamine-rl # etils - # keras + # labmaze # ml-collections + # mujoco # open-spiel + # optax + # orbax-checkpoint + # recsim # tensorboard # tensorflow # tensorflow-datasets # tensorflow-metadata # tensorflow-probability + # tf-slim accelerate==0.28.0 # via -r /ray/ci/../python/requirements/ml/core-requirements.txt adagio==0.2.4 - # via fugue + # via + # fugue + # qpd adal==1.2.7 # via msrestazure +aim==3.17.5 + # via -r /ray/ci/../python/requirements/ml/tune-test-requirements.txt +aim-ui==3.17.5 + # via aim +aimrecords==0.0.7 + # via aim +aimrocks==0.4.0 + # via aim aioboto3==11.2.0 # via -r /ray/ci/../python/requirements/ml/data-requirements.txt aiobotocore==2.5.0 @@ -27,9 +46,10 @@ aiobotocore==2.5.0 # s3fs aiofiles==22.1.0 # via + # aim # gradio # ypy-websocket -aiohttp==3.9.5 ; python_version >= "3.12" +aiohttp==3.9.5 # via # -r /ray/ci/../python/requirements.txt # aiobotocore @@ -49,18 +69,28 @@ aiosignal==1.3.1 # via # -r /ray/ci/../python/requirements.txt # aiohttp -aiosqlite==0.20.0 + # ray +aiosqlite==0.19.0 # via ypy-websocket alabaster==0.7.13 # via sphinx -alembic==1.13.1 +ale-py==0.8.1 # via + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt + # gym +alembic==1.12.1 + # via + # aim # mlflow # optuna altair==5.1.2 # via gradio annotated-types==0.6.0 # via pydantic +antlr4-python3-runtime==4.11.1 + # via + # fugue-sql-antlr + # qpd anyio==3.7.1 # via # httpcore @@ -71,7 +101,7 @@ appdirs==1.4.4 # via fs applicationinsights==0.11.10 # via azure-cli-telemetry -argcomplete==1.12.3 +argcomplete==3.3.0 # via # azure-cli-core # gsutil @@ -80,15 +110,20 @@ argcomplete==1.12.3 argon2-cffi==23.1.0 # via # jupyter-server + # nbclassic # notebook argon2-cffi-bindings==21.2.0 # via argon2-cffi -array-record==0.5.1 ; sys_platform != "darwin" and platform_system != "Windows" +argparse==1.4.0 + # via mosaicml +array-record==0.5.1 ; python_version < "3.12" and sys_platform != "darwin" and platform_system != "Windows" # via # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt # tensorflow-datasets arrow==1.3.0 # via isoduration +asn1crypto==1.5.1 + # via snowflake-connector-python asttokens==2.4.1 # via stack-data astunparse==1.6.3 @@ -122,9 +157,9 @@ aws-xray-sdk==2.12.1 # via moto ax-platform==0.3.2 # via -r /ray/ci/../python/requirements/ml/tune-requirements.txt -azure-cli-core==2.43.0 +azure-cli-core==2.62.0 # via -r /ray/ci/../python/requirements/test-requirements.txt -azure-cli-telemetry==1.0.8 +azure-cli-telemetry==1.1.0 # via azure-cli-core azure-common==1.1.28 # via @@ -136,7 +171,7 @@ azure-core==1.29.5 # azure-identity # azure-mgmt-core # msrest -azure-identity==1.14.1 +azure-identity==1.17.1 # via -r /ray/ci/../python/requirements/test-requirements.txt azure-mgmt-compute==31.0.0 # via -r /ray/ci/../python/requirements/test-requirements.txt @@ -161,11 +196,13 @@ backoff==1.10.0 # -r /ray/ci/../python/requirements/test-requirements.txt # opentelemetry-exporter-otlp-proto-grpc # segment-analytics-python +base58==2.0.1 + # via aimrecords bayesian-optimization==1.4.3 # via # -r /ray/ci/../python/requirements/ml/tune-requirements.txt # nevergrad -bcrypt==4.1.3 +bcrypt==4.0.1 # via paramiko beautifulsoup4==4.11.1 # via @@ -204,8 +241,12 @@ braceexpand==0.1.7 # via webdataset bracex==2.4 # via wcmatch +cached-property==1.5.2 + # via orbax-checkpoint cachetools==5.3.2 - # via google-auth + # via + # aim + # google-auth certifi==2023.11.17 # via # httpcore @@ -214,24 +255,31 @@ certifi==2023.11.17 # msrest # requests # sentry-sdk + # snowflake-connector-python cffi==1.16.0 # via # argon2-cffi-bindings # cryptography # pymunk # pynacl + # snowflake-connector-python # soundfile cfn-lint==0.83.3 # via moto charset-normalizer==3.3.2 - # via requests + # via + # requests + # snowflake-connector-python chess==1.7.0 # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt +chex==0.1.7 + # via optax clang-format==12.0.1 # via -r /ray/ci/../python/requirements/lint-requirements.txt click==8.1.7 # via # -r /ray/ci/../python/requirements.txt + # aim # autorom # autorom-accept-rom-license # black @@ -242,6 +290,7 @@ click==8.1.7 # flask # jupyter-cache # mlflow + # ray # semgrep # tensorflow-datasets # typer @@ -254,17 +303,19 @@ cloudpickle==2.2.0 # -r /ray/ci/../python/requirements/test-requirements.txt # dask # distributed + # gym # gymnasium # hyperopt # mlagents-envs # mlflow + # pymars # statsforecast # tensorflow-probability -cma==3.3.0 +cma==3.2.2 # via nevergrad cmaes==0.10.0 # via optuna -cmdstanpy==1.2.3 +cmdstanpy==1.2.0 # via prophet colorama==0.4.6 # via @@ -275,7 +326,7 @@ coloredlogs==15.0.1 # via onnxruntime colorful==0.5.5 # via -r /ray/ci/../python/requirements.txt -colorlog==6.8.2 +colorlog==6.7.0 # via optuna comet-ml==3.31.9 # via -r /ray/ci/../python/requirements/ml/core-requirements.txt @@ -292,11 +343,11 @@ configspace==0.7.1 # -r /ray/ci/../python/requirements/ml/tune-requirements.txt # hpbandster contextlib2==21.6.0 - # via ml-collections + # via + # ml-collections + # pytest-shutil contourpy==1.1.1 # via matplotlib -cramjam==2.8.3 - # via python-snappy crc32c==2.3 # via -r /ray/ci/../python/requirements/ml/data-requirements.txt crcmod==1.7 @@ -305,6 +356,7 @@ cryptography==38.0.1 # via # -r /ray/ci/../python/requirements/test-requirements.txt # adal + # aim # azure-cli-core # azure-identity # moto @@ -313,6 +365,7 @@ cryptography==38.0.1 # pyjwt # pyopenssl # python-jose + # snowflake-connector-python # sshpubkeys # trustme cupy-cuda12x==13.1.0 ; sys_platform != "darwin" @@ -323,16 +376,18 @@ cython==0.29.37 # via # -r /ray/ci/../python/requirements/test-requirements.txt # gpy -dask==2022.10.1; python_version < '3.12' -dask==2024.6.0; python_version >= '3.12' - # via dask +dask==2022.10.1 ; python_version < "3.12" + # via + # -r /ray/ci/../python/requirements/ml/data-requirements.txt + # distributed databricks-cli==0.18.0 # via mlflow -datasets==2.18.0 +datasets==2.14.0 # via # -r /ray/ci/../python/requirements/ml/data-test-requirements.txt # -r /ray/ci/../python/requirements/ml/train-requirements.txt # evaluate + # mosaicml debugpy==1.8.0 # via ipykernel decorator==5.1.1 @@ -341,25 +396,36 @@ decorator==5.1.1 # moviepy # paramz # tensorflow-probability +decord==0.6.0 + # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt deepspeed==0.12.3 # via -r /ray/ci/../python/requirements/ml/train-requirements.txt defusedxml==0.7.1 # via # nbconvert + # pymars # semgrep -dill==0.3.8 +dill==0.3.7 # via # datasets # evaluate # multiprocess distlib==0.3.7 # via virtualenv -distributed==2022.10.1; python_version < '3.12' -distributed==2024.6.0; python_version >= '3.12' +distributed==2022.10.1 # via dask +distro==1.9.0 + # via azure-cli-core +dm-control==1.0.12 + # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt +dm-env==1.6 + # via dm-control dm-tree==0.1.8 # via # -r /ray/ci/../python/requirements.txt + # chex + # dm-control + # dm-env # tensorflow-datasets # tensorflow-probability dnspython==2.4.2 @@ -371,12 +437,18 @@ docker==6.1.3 # pytest-docker-tools docker-pycreds==0.4.0 # via wandb +docstring-parser==0.15 + # via yahp docutils==0.19 # via # -r /ray/ci/../python/requirements/lint-requirements.txt # myst-parser # sphinx -dulwich==0.22.1 +dopamine-rl==4.0.5 ; sys_platform != "darwin" or platform_machine != "arm64" + # via + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt + # recsim +dulwich==0.21.6 # via comet-ml ecdsa==0.18.0 # via @@ -387,17 +459,21 @@ entrypoints==0.4 # via # jupyter-client # mlflow + # nbconvert et-xmlfile==1.1.0 # via openpyxl -etils==1.5.2 +etils==1.5.2 ; python_version < "3.12" # via # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt # array-record + # orbax-checkpoint # tensorflow-datasets evaluate==0.4.0 # via -r /ray/ci/../python/requirements/ml/train-test-requirements.txt everett==3.3.0 # via comet-ml +execnet==2.1.1 + # via pytest-shutil executing==2.0.1 # via stack-data face==22.0.0 @@ -410,6 +486,7 @@ fastapi==0.109.2 # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/test-requirements.txt + # aim # gradio fastavro==1.9.4 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt @@ -428,8 +505,10 @@ ffmpy==0.3.1 filelock==3.13.1 # via # -r /ray/ci/../python/requirements.txt - # datasets + # aim # huggingface-hub + # ray + # snowflake-connector-python # torch # transformers # virtualenv @@ -460,6 +539,8 @@ flatbuffers==23.5.26 # onnxruntime # tensorflow # tf2onnx +flax==0.7.2 + # via dopamine-rl fonttools==4.45.1 # via matplotlib fqdn==1.5.1 @@ -471,6 +552,7 @@ frozenlist==1.4.0 # -r /ray/ci/../python/requirements.txt # aiohttp # aiosignal + # ray fs==2.4.16 # via triad fsspec==2023.5.0 @@ -484,26 +566,37 @@ fsspec==2023.5.0 # huggingface-hub # modin # pytorch-lightning + # ray # s3fs # torch # torch-geometric # triad -fugue==0.9.0 +fugue==0.8.7 # via statsforecast -future==1.0.0 +fugue-sql-antlr==0.2.0 + # via fugue +future==0.18.3 # via hyperopt gast==0.4.0 # via # tensorflow # tensorflow-probability -gcs-oauth2-boto-plugin==3.1 +gcs-oauth2-boto-plugin==3.0 # via gsutil +gin-config==0.5.0 + # via + # dopamine-rl + # recsim gitdb==4.0.11 # via gitpython -gitpython==3.1.43 +gitpython==3.1.40 # via # mlflow # wandb +glfw==2.6.3 + # via + # dm-control + # mujoco glom==22.1.0 # via semgrep google-api-core==1.34.0 @@ -524,13 +617,17 @@ google-auth==2.23.4 # google-api-core # google-api-python-client # google-auth-httplib2 + # google-auth-oauthlib # google-cloud-core # google-cloud-storage # gsutil # kubernetes + # tensorboard google-auth-httplib2==0.1.1 # via google-api-python-client -google-cloud-bigquery==3.18.0 +google-auth-oauthlib==1.0.0 + # via tensorboard +google-cloud-bigquery==3.13.0 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt google-cloud-bigquery-storage==2.24.0 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt @@ -579,7 +676,9 @@ grpcio==1.60.0 ; sys_platform != "darwin" # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/anyscale-requirements.txt + # aim # google-api-core + # google-cloud-bigquery # grpcio-status # mlagents-envs # opencensus-proto @@ -591,14 +690,22 @@ grpcio-status==1.48.2 # via google-api-core gsutil==5.27 # via -r /ray/ci/../python/requirements/docker/ray-docker-requirements.txt -gunicorn==21.2.0 +gunicorn==20.1.0 # via mlflow +gym==0.26.2 + # via + # dopamine-rl + # recsim +gym-notices==0.0.8 + # via gym gymnasium==0.28.1 # via # -r /ray/ci/../python/requirements.txt + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt # minigrid # pettingzoo # shimmy + # supersuit h11==0.12.0 # via # -r /ray/ci/../python/requirements/test-requirements.txt @@ -607,7 +714,6 @@ h11==0.12.0 h5py==3.10.0 # via # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt - # keras # tensorflow higher==0.2.1 # via -r /ray/ci/../python/requirements/ml/rllib-requirements.txt @@ -659,10 +765,12 @@ idna==3.7 # httpx # jsonschema # requests + # snowflake-connector-python # trustme # yarl -imageio==2.34.2 +imageio==2.31.1 # via + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt # moviepy # scikit-image imageio-ffmpeg==0.4.5 @@ -677,8 +785,10 @@ importlib-metadata==6.11.0 # myst-nb importlib-resources==5.13.0 # via + # ale-py # etils # gradio + # orbax-checkpoint # prophet iniconfig==2.0.0 # via pytest @@ -716,19 +826,35 @@ isort==5.10.1 # via -r /ray/ci/../python/requirements/lint-requirements.txt itsdangerous==2.1.2 # via flask +jax==0.4.13 + # via + # chex + # dopamine-rl + # flax + # optax + # orbax-checkpoint jax-jumpy==1.0.0 # via gymnasium +jaxlib==0.4.13 + # via + # chex + # dopamine-rl + # optax + # orbax-checkpoint jedi==0.19.1 # via ipython jinja2==3.1.2 # via # -r /ray/ci/../python/requirements/test-requirements.txt + # aim # altair # ax-platform # bokeh # dask # distributed # flask + # fugue + # fugue-sql-antlr # gradio # jupyter-server # jupyterlab @@ -737,6 +863,7 @@ jinja2==3.1.2 # mlflow # moto # myst-parser + # nbclassic # nbconvert # notebook # sphinx @@ -754,7 +881,7 @@ joblib==1.2.0 # scikit-learn jschema-to-python==1.2.3 # via cfn-lint -json5==0.9.25 +json5==0.9.14 # via jupyterlab-server jsondiff==2.0.0 # via moto @@ -782,17 +909,19 @@ jsonschema==4.17.3 # nbformat # openapi-schema-validator # openapi-spec-validator + # ray # semgrep jsonschema-spec==0.1.6 # via openapi-spec-validator junit-xml==1.9 # via cfn-lint -jupyter-cache==1.0.0 +jupyter-cache==0.6.1 # via myst-nb jupyter-client==7.3.4 # via # ipykernel # jupyter-server + # nbclassic # nbclient # notebook jupyter-core==5.5.0 @@ -801,6 +930,7 @@ jupyter-core==5.5.0 # jupyter-client # jupyter-server # jupyterlab + # nbclassic # nbconvert # nbformat # notebook @@ -811,8 +941,9 @@ jupyter-server==1.24.0 # jupyter-server-fileid # jupyterlab # jupyterlab-server + # nbclassic # notebook-shim -jupyter-server-fileid==0.9.2 +jupyter-server-fileid==0.9.0 # via jupyter-server-ydoc jupyter-server-ydoc==0.6.1 # via jupyterlab @@ -834,15 +965,17 @@ jupytext==1.13.6 # via -r /ray/ci/../python/requirements/test-requirements.txt kaggle-environments==1.7.11 # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt -keras==3.3.3 +keras==2.15.0 # via tensorflow kiwisolver==1.4.5 # via matplotlib -knack==0.10.1 +knack==0.11.0 # via azure-cli-core kubernetes==24.2.0 # via -r /ray/ci/../python/requirements/test-requirements.txt -lazy-loader==0.4 +labmaze==1.0.6 + # via dm-control +lazy-loader==0.3 # via scikit-image lazy-object-proxy==1.9.0 # via openapi-spec-validator @@ -867,12 +1000,12 @@ locket==1.0.0 lxml==4.9.4 # via # -r /ray/ci/../python/requirements/test-requirements.txt + # dm-control + # nbconvert # vsphere-automation-sdk lz4==4.3.2 - # via - # -r /ray/ci/../python/requirements.txt - # dask -mako==1.3.5 + # via -r /ray/ci/../python/requirements.txt +mako==1.3.0 # via alembic markdown==3.5.1 # via @@ -908,44 +1041,53 @@ mdit-py-plugins==0.3.5 # via # jupytext # myst-parser +medpy==0.4.0 + # via mosaicml memray==1.10.0 ; platform_system != "Windows" and sys_platform != "darwin" and platform_machine != "aarch64" # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/test-requirements.txt minigrid==2.3.1 # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt -mistune==3.0.2 +mistune==0.8.4 # via nbconvert ml-collections==0.1.1 # via open-spiel ml-dtypes==0.3.2 # via - # keras + # jax + # jaxlib # tensorflow mlagents-envs==0.28.0 # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt mlflow==2.9.2 # via -r /ray/ci/../python/requirements/ml/core-requirements.txt -modin==0.30.1 +mock==5.1.0 + # via pytest-shutil +modin==0.22.2 # via -r /ray/ci/../python/requirements/ml/data-requirements.txt +monai==1.3.2 + # via mosaicml monotonic==1.6 # via # gsutil # segment-analytics-python -more-itertools==10.2.0 +more-itertools==10.1.0 # via configspace +mosaicml==0.2.4 + # via -r /ray/ci/../python/requirements/ml/train-test-requirements.txt moto==4.2.12 # via -r /ray/ci/../python/requirements/test-requirements.txt moviepy==0.2.3.1 # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt mpmath==1.3.0 # via sympy -msal==1.20.0 +msal==1.28.1 # via # azure-cli-core # azure-identity # msal-extensions -msal-extensions==1.0.0 +msal-extensions==1.2.0b1 # via # azure-cli-core # azure-identity @@ -954,7 +1096,10 @@ msgpack==1.0.7 # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/ml/rllib-requirements.txt # distributed + # flax # msgpack-numpy + # orbax-checkpoint + # ray msgpack-numpy==0.4.8 # via -r /ray/ci/../python/requirements/ml/rllib-requirements.txt msrest==0.7.1 @@ -963,13 +1108,17 @@ msrestazure==0.6.4 # via # -r /ray/ci/../python/requirements/test-requirements.txt # azure-cli-core +mujoco==2.3.6 + # via + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt + # dm-control multidict==6.0.4 # via # aiohttp # yarl multipledispatch==1.0.0 # via botorch -multiprocess==0.70.16 +multiprocess==0.70.15 # via # datasets # evaluate @@ -987,9 +1136,7 @@ myst-parser==1.0.0 # via # -r /ray/ci/../python/requirements/test-requirements.txt # myst-nb -namex==0.0.8 - # via keras -nbclassic==1.1.0 +nbclassic==1.0.0 # via # jupyterlab # notebook @@ -998,9 +1145,10 @@ nbclient==0.5.13 # jupyter-cache # myst-nb # nbconvert -nbconvert==7.16.4 +nbconvert==6.5.4 # via # jupyter-server + # nbclassic # notebook nbformat==5.9.2 # via @@ -1008,6 +1156,7 @@ nbformat==5.9.2 # jupyter-server # jupytext # myst-nb + # nbclassic # nbclient # nbconvert # notebook @@ -1018,8 +1167,11 @@ nest-asyncio==1.5.8 # nbclassic # nbclient # notebook + # orbax-checkpoint netifaces==0.11.0 - # via hpbandster + # via + # hpbandster + # raydp networkx==3.2.1 # via # cfn-lint @@ -1032,7 +1184,7 @@ ninja==1.11.1.1 # via deepspeed notebook==6.5.7 # via jupyterlab -notebook-shim==0.2.4 +notebook-shim==0.2.3 # via nbclassic # via vsphere-automation-sdk # via vsphere-automation-sdk @@ -1043,15 +1195,20 @@ numba==0.59.1 # -r /ray/ci/../python/requirements/test-requirements.txt # statsforecast numexpr==2.8.4 - # via -r /ray/ci/../python/requirements/test-requirements.txt + # via + # -r /ray/ci/../python/requirements/test-requirements.txt + # pymars numpy==1.26.4 # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/test-requirements.txt # accelerate + # aim + # ale-py # altair # bayesian-optimization # bokeh + # chex # cma # cmaes # cmdstanpy @@ -1060,35 +1217,49 @@ numpy==1.26.4 # cupy-cuda12x # dask # datasets + # decord # deepspeed + # dm-control + # dm-env + # dopamine-rl # etils # evaluate + # flax # gpy # gradio + # gym # gymnasium # h5py # hpbandster # hyperopt # imageio + # jax # jax-jumpy - # keras + # jaxlib + # labmaze # lightgbm # matplotlib + # medpy # minigrid # ml-dtypes # mlagents-envs # mlflow # modin + # monai # moviepy # msgpack-numpy + # mujoco # nevergrad # numba # numexpr # onnx # onnxruntime # open-spiel + # opencv-python # opt-einsum + # optax # optuna + # orbax-checkpoint # pandas # paramz # patsy @@ -1096,8 +1267,12 @@ numpy==1.26.4 # prophet # pyarrow # pylance + # pymars # pyro-ppl # pytorch-lightning + # pywavelets + # raydp + # recsim # scikit-image # scikit-learn # scipy @@ -1105,19 +1280,23 @@ numpy==1.26.4 # stanio # statsforecast # statsmodels + # supersuit # tensorboard # tensorboardx # tensorflow # tensorflow-datasets # tensorflow-probability + # tensorstore # tf2onnx # tifffile + # tinyscaler # torch-geometric # torchmetrics # torchtext # torchvision # transformers # triad + # tune-sklearn # utilsforecast # webdataset # xgboost @@ -1148,6 +1327,8 @@ opencensus-context==0.1.3 # via opencensus opencensus-proto==0.1.0 # via opentelemetry-exporter-opencensus +opencv-python==4.8.1.78 + # via dopamine-rl openpyxl==3.0.10 # via -r /ray/ci/../python/requirements/test-requirements.txt opentelemetry-api==1.1.0 @@ -1180,18 +1361,22 @@ opentelemetry-semantic-conventions==0.20b0 # via opentelemetry-sdk opt-einsum==3.3.0 # via + # jax # pyro-ppl # tensorflow -optree==0.11.0 - # via keras +optax==0.1.7 + # via flax optuna==3.2.0 # via -r /ray/ci/../python/requirements/ml/tune-requirements.txt +orbax-checkpoint==0.2.3 + # via flax orjson==3.9.10 # via gradio -packaging==21.3 +packaging==23.0 # via # -r /ray/ci/../python/requirements.txt # accelerate + # aim # altair # azure-cli-core # bokeh @@ -1201,16 +1386,16 @@ packaging==21.3 # distributed # docker # evaluate + # fugue-sql-antlr # google-cloud-bigquery # gradio # gradio-client - # gunicorn # huggingface-hub # ipykernel # jupyter-server # jupyterlab # jupyterlab-server - # lazy-loader + # knack # lightning-utilities # matplotlib # mlflow @@ -1223,8 +1408,10 @@ packaging==21.3 # pytest-rerunfailures # pytest-sugar # pytorch-lightning + # ray # scikit-image # semgrep + # snowflake-connector-python # sphinx # statsmodels # tensorboardx @@ -1232,7 +1419,7 @@ packaging==21.3 # torchmetrics # transformers # utilsforecast -pandas==2.2.2 +pandas==1.5.3 # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/ml/data-requirements.txt @@ -1241,16 +1428,21 @@ pandas==2.2.2 # cmdstanpy # dask # datasets + # dopamine-rl # evaluate # gradio # mlflow # modin # prophet + # pymars + # qpd + # ray + # raydp # statsforecast # statsmodels # triad # utilsforecast -pandocfilters==1.5.1 +pandocfilters==1.5.0 # via nbconvert paramiko==2.12.0 # via azure-cli-core @@ -1258,13 +1450,17 @@ paramz==0.9.6 # via gpy parso==0.8.3 # via jedi -partd==1.4.2 +partd==1.4.1 # via dask +path==16.14.0 + # via path-py +path-py==12.5.0 + # via pytest-shutil pathable==0.4.3 # via jsonschema-spec pathspec==0.11.2 # via black -patsy==0.5.6 +patsy==0.5.3 # via statsmodels pbr==6.0.0 # via @@ -1283,24 +1479,27 @@ pickleshare==0.7.5 pillow==10.3.0 ; platform_system != "Windows" # via # -r /ray/ci/../python/requirements/test-requirements.txt + # aim # bokeh + # dopamine-rl # gradio # imageio # matplotlib # mlagents-envs # scikit-image # torchvision -pkginfo==1.11.1 +pkginfo==1.9.6 # via azure-cli-core platformdirs==3.11.0 # via # black # jupyter-core + # snowflake-connector-python # virtualenv # wandb -plotly==5.22.0 +plotly==5.18.0 # via ax-platform -pluggy==1.5.0 +pluggy==1.3.0 # via pytest polars==0.14.21 # via -r /ray/ci/../python/requirements/test-requirements.txt @@ -1312,6 +1511,7 @@ prometheus-client==0.19.0 # via # -r /ray/ci/../python/requirements.txt # jupyter-server + # nbclassic # notebook promise==2.3 # via tensorflow-datasets @@ -1319,12 +1519,17 @@ prompt-toolkit==3.0.41 # via ipython prophet==1.1.5 # via -r /ray/ci/../python/requirements/ml/tune-test-requirements.txt -proto-plus==1.23.0 - # via google-cloud-bigquery-storage +proto-plus==1.22.3 + # via + # google-cloud-bigquery + # google-cloud-bigquery-storage protobuf==3.20.3 # via # -r /ray/ci/../python/requirements.txt + # aim + # dm-control # google-api-core + # google-cloud-bigquery # google-cloud-bigquery-storage # googleapis-common-protos # grpcio-status @@ -1335,6 +1540,8 @@ protobuf==3.20.3 # opentelemetry-exporter-opencensus # opentelemetry-proto # proto-plus + # ray + # raydp # tensorboard # tensorboardx # tensorflow @@ -1347,11 +1554,14 @@ proxy-py==2.4.3 psutil==5.9.6 # via # accelerate + # aim # azure-cli-core # deepspeed # distributed # ipykernel # modin + # pymars + # raydp # tensorflow-datasets # torch-geometric # wandb @@ -1369,23 +1579,22 @@ py-partiql-parser==0.5.0 # via moto py-spy==0.3.14 # via -r /ray/ci/../python/requirements.txt +py3nvml==0.2.7 + # via aim py4j==0.10.9.7 # via # hyperopt # pyspark -pyarrow==14.0.2 ; python_version >= "3.12" +pyarrow==14.0.2 # via # -r /ray/ci/../python/requirements.txt - # dask # datasets # feather-format # mlflow # pylance + # ray + # raydp # triad -pyarrow-hotfix==0.6 - # via - # dask - # datasets pyasn1==0.5.1 # via # oauth2client @@ -1419,6 +1628,7 @@ pyflakes==2.3.1 pygame==2.5.2 # via # -r /ray/ci/../python/requirements/test-requirements.txt + # dopamine-rl # minigrid pyglet==1.5.15 # via -r /ray/ci/../python/requirements/ml/rllib-requirements.txt @@ -1436,8 +1646,11 @@ pyjwt==2.8.0 # azure-cli-core # databricks-cli # msal -pylance==0.12.1 + # snowflake-connector-python +pylance==0.10.18 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt +pymars==0.10.0 + # via -r /ray/ci/../python/requirements/ml/data-requirements.txt pymongo==4.3.2 # via -r /ray/ci/../python/requirements/test-requirements.txt pymunk==6.2.1 @@ -1446,6 +1659,10 @@ pynacl==1.5.0 # via paramiko pynvml==11.5.0 # via deepspeed +pyopengl==3.1.7 + # via + # dm-control + # mujoco pyopenssl==23.0.0 # via # -r /ray/ci/../python/requirements/anyscale-requirements.txt @@ -1453,19 +1670,20 @@ pyopenssl==23.0.0 # azure-cli-core # gcs-oauth2-boto-plugin # gsutil + # snowflake-connector-python # vapi-runtime pyparsing==3.1.1 # via # configspace + # dm-control # httplib2 # matplotlib # moto - # packaging # pydot # torch-geometric pyro-api==0.1.2 # via pyro-ppl -pyro-ppl==1.9.1 +pyro-ppl==1.8.6 # via botorch pyro4==4.82 # via hpbandster @@ -1474,24 +1692,31 @@ pyrsistent==0.20.0 pysocks==1.7.1 # via requests pyspark==3.4.1 - # via -r /ray/ci/../python/requirements/test-requirements.txt + # via + # -r /ray/ci/../python/requirements/test-requirements.txt + # raydp pytest==7.4.4 # via # -r /ray/ci/../python/requirements/test-requirements.txt # pytest-asyncio # pytest-docker-tools + # pytest-fixture-config # pytest-forked # pytest-lazy-fixture # pytest-remotedata # pytest-repeat # pytest-rerunfailures + # pytest-shutil # pytest-sphinx # pytest-sugar # pytest-timeout + # pytest-virtualenv pytest-asyncio==0.16.0 # via -r /ray/ci/../python/requirements/test-requirements.txt pytest-docker-tools==3.1.3 # via -r /ray/ci/../python/requirements/test-requirements.txt +pytest-fixture-config==1.7.0 + # via pytest-virtualenv pytest-forked==1.4.0 # via -r /ray/ci/../python/requirements/test-requirements.txt pytest-httpserver==1.0.6 @@ -1504,15 +1729,20 @@ pytest-repeat==0.9.3 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt pytest-rerunfailures==11.1.2 # via -r /ray/ci/../python/requirements/test-requirements.txt +pytest-shutil==1.7.0 + # via pytest-virtualenv pytest-sphinx @ git+https://github.com/ray-project/pytest-sphinx # via -r /ray/ci/../python/requirements/test-requirements.txt pytest-sugar==0.9.5 # via -r /ray/ci/../python/requirements/test-requirements.txt pytest-timeout==2.1.0 # via -r /ray/ci/../python/requirements/test-requirements.txt +pytest-virtualenv==1.7.0 ; python_version < "3.12" + # via -r /ray/ci/../python/requirements/test-requirements.txt python-dateutil==2.8.2 # via # adal + # aim # arrow # botocore # freezegun @@ -1532,19 +1762,23 @@ python-lsp-jsonrpc==1.0.0 # via semgrep python-multipart==0.0.6 # via gradio -python-snappy==0.7.1 - # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt pytorch-lightning==1.8.6 # via -r /ray/ci/../python/requirements/ml/tune-test-requirements.txt +pytorch-ranger==0.1.1 + # via torch-optimizer pytz==2022.7.1 # via # -r /ray/ci/../python/requirements/test-requirements.txt + # aim # mlflow # pandas + # snowflake-connector-python pyu2f==0.1.5 # via google-reauth pyvmomi==8.0.2.0.1 # via vsphere-automation-sdk +pywavelets==1.5.0 + # via scikit-image pyyaml==6.0.1 # via # -r /ray/ci/../python/requirements.txt @@ -1555,6 +1789,7 @@ pyyaml==6.0.1 # dask # datasets # distributed + # flax # gradio # huggingface-hub # jsonschema-spec @@ -1566,15 +1801,20 @@ pyyaml==6.0.1 # ml-collections # mlagents-envs # mlflow + # mosaicml # moto # myst-nb # myst-parser # optuna + # orbax-checkpoint + # pymars # pytorch-lightning + # ray # timm # transformers # wandb # webdataset + # yahp # yq pyzmq==26.0.3 # via @@ -1582,9 +1822,19 @@ pyzmq==26.0.3 # ipykernel # jupyter-client # jupyter-server + # nbclassic # notebook +qpd==0.4.4 + # via fugue querystring-parser==1.2.4 # via mlflow + # via + # raydp + # tune-sklearn +raydp==1.7.0b20231020.dev0 + # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt +recsim==0.2.4 ; sys_platform != "darwin" or platform_machine != "arm64" + # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt redis==4.4.2 # via -r /ray/ci/../python/requirements/test-requirements.txt regex==2023.10.3 @@ -1595,6 +1845,7 @@ requests==2.31.0 # via # -r /ray/ci/../python/requirements.txt # adal + # aim # autorom # autorom-accept-rom-license # azure-cli-core @@ -1602,6 +1853,7 @@ requests==2.31.0 # comet-ml # databricks-cli # datasets + # dm-control # docker # evaluate # fsspec @@ -1619,13 +1871,15 @@ requests==2.31.0 # moto # msal # msrest + # ray # requests-oauthlib # requests-toolbelt # responses # segment-analytics-python # semgrep + # snowflake-connector-python # sphinx - # tensorflow + # tensorboard # tensorflow-datasets # tf2onnx # torch-geometric @@ -1635,6 +1889,7 @@ requests==2.31.0 # wandb requests-oauthlib==1.3.1 # via + # google-auth-oauthlib # kubernetes # msrest requests-toolbelt==1.0.0 @@ -1644,6 +1899,8 @@ responses==0.13.4 # -r /ray/ci/../python/requirements/ml/data-requirements.txt # evaluate # moto +restrictedpython==7.0 + # via aim retry-decorator==1.1.1 # via # gcs-oauth2-boto-plugin @@ -1661,7 +1918,7 @@ rich==12.6.0 # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/ml/rllib-requirements.txt - # keras + # flax # memray # semgrep rsa==4.7.2 @@ -1671,21 +1928,23 @@ rsa==4.7.2 # oauth2client # python-jose ruamel-yaml==0.17.40 - # via semgrep + # via + # semgrep + # yahp ruamel-yaml-clib==0.2.8 # via ruamel-yaml s3fs==2023.5.0 # via -r /ray/ci/../python/requirements/ml/core-requirements.txt s3transfer==0.6.2 # via boto3 -safetensors==0.4.3 +safetensors==0.4.1 # via # accelerate # timm # transformers sarif-om==1.0.4 # via cfn-lint -scikit-image==0.24.0 +scikit-image==0.21.0 # via -r /ray/ci/../python/requirements.txt scikit-learn==1.3.2 # via @@ -1695,7 +1954,10 @@ scikit-learn==1.3.2 # gpytorch # lightgbm # mlflow + # mosaicml + # pymars # torch-geometric + # tune-sklearn scipy==1.11.4 # via # -r /ray/ci/../python/requirements.txt @@ -1703,14 +1965,20 @@ scipy==1.11.4 # bayesian-optimization # botorch # configspace + # dm-control # gpy # hpbandster # hyperopt + # jax + # jaxlib # lightgbm # linear-operator + # medpy # mlflow # open-spiel # paramz + # pymars + # recsim # scikit-image # scikit-learn # statsforecast @@ -1718,22 +1986,26 @@ scipy==1.11.4 # torch-cluster # torch-geometric # torch-sparse + # tune-sklearn # xgboost segment-analytics-python==2.2.0 - # via -r /ray/ci/../python/requirements/test-requirements.txt + # via + # -r /ray/ci/../python/requirements/test-requirements.txt + # aim semantic-version==2.10.0 # via # comet-ml # gradio semgrep==1.32.0 # via -r /ray/ci/../python/requirements/lint-requirements.txt -send2trash==1.8.3 +send2trash==1.8.2 # via # jupyter-server + # nbclassic # notebook sentencepiece==0.1.96 # via -r /ray/ci/../python/requirements/ml/train-test-requirements.txt -sentry-sdk==2.5.0 +sentry-sdk==1.37.1 # via # comet-ml # wandb @@ -1749,6 +2021,8 @@ shimmy==1.3.0 # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt shortuuid==1.0.1 # via -r /ray/ci/../python/requirements/ml/tune-test-requirements.txt +simpleitk==2.3.1 + # via medpy six==1.16.0 # via # asttokens @@ -1778,6 +2052,7 @@ six==1.16.0 # patsy # promise # pytest-remotedata + # pytest-shutil # python-dateutil # pyu2f # pyvmomi @@ -1803,8 +2078,12 @@ sniffio==1.3.0 # httpx snowballstemmer==2.2.0 # via sphinx +snowflake-connector-python==3.11.0 + # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt sortedcontainers==2.4.0 - # via distributed + # via + # distributed + # snowflake-connector-python soundfile==0.12.1 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt soupsieve==2.5 @@ -1828,18 +2107,22 @@ sphinxcontrib-serializinghtml==1.1.5 # via sphinx sqlalchemy==1.4.17 # via + # aim # alembic # ax-platform # jupyter-cache # mlflow # optuna -sqlparse==0.5.0 + # pymars +sqlglot==20.4.0 + # via fugue +sqlparse==0.4.4 # via mlflow sshpubkeys==3.3.1 # via moto stack-data==0.6.3 # via ipython -stanio==0.5.0 +stanio==0.3.0 # via cmdstanpy starlette==0.36.3 # via @@ -1847,10 +2130,12 @@ starlette==0.36.3 # fastapi statsforecast==1.7.0 # via -r /ray/ci/../python/requirements/ml/tune-test-requirements.txt -statsmodels==0.14.2 +statsmodels==0.14.0 # via # hpbandster # statsforecast +supersuit==3.8.0 + # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt sympy==1.12 # via # cfn-lint @@ -1865,9 +2150,9 @@ tblib==3.0.0 # via # -r /ray/ci/../python/requirements/docker/ray-docker-requirements.txt # distributed -tenacity==8.3.0 +tenacity==8.2.3 # via plotly -tensorboard==2.16.2 +tensorboard==2.15.2 # via # -r /ray/ci/../python/requirements/test-requirements.txt # tensorflow @@ -1880,34 +2165,49 @@ tensorboardx==2.6.2.2 # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/test-requirements.txt # pytorch-lightning -tensorflow==2.16.1 ; sys_platform != "darwin" or platform_machine != "arm64" + # ray +tensorflow==2.15.1 ; python_version < "3.12" and (sys_platform != "darwin" or platform_machine != "arm64") # via # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt - # tensorflow-probability - # tf-keras -tensorflow-datasets==4.9.3 + # dopamine-rl + # recsim +tensorflow-datasets==4.9.3 ; python_version < "3.12" # via # -r /ray/ci/../python/requirements/ml/data-test-requirements.txt # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt tensorflow-estimator==2.15.0 - # via -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt + # via + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt + # tensorflow +tensorflow-io-gcs-filesystem==0.31.0 ; python_version < "3.12" + # via + # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt + # tensorflow tensorflow-metadata==1.13.0 # via tensorflow-datasets -tensorflow-probability==0.24.0 - # via -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt +tensorflow-probability==0.23.0 ; python_version < "3.12" + # via + # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt + # dopamine-rl +tensorstore==0.1.45 + # via + # flax + # orbax-checkpoint termcolor==2.3.0 # via + # pytest-shutil # pytest-sugar # tensorflow # tensorflow-datasets -terminado==0.18.1 +terminado==0.18.0 # via # jupyter-server + # nbclassic # notebook testfixtures==7.0.0 # via -r /ray/ci/../python/requirements/test-requirements.txt -tf-keras==2.16.0 - # via tensorflow-probability +tf-slim==1.1.0 + # via dopamine-rl tf2onnx==1.15.1 ; sys_platform != "darwin" or platform_machine != "arm64" # via -r /ray/ci/../python/requirements/ml/rllib-requirements.txt threadpoolctl==3.1.0 @@ -1918,8 +2218,12 @@ tifffile==2023.7.10 # via scikit-image timm==0.9.2 # via -r /ray/ci/../python/requirements/ml/tune-test-requirements.txt -tinycss2==1.3.0 +tinycss2==1.2.1 # via nbconvert +tinyscaler==1.2.6 + # via + # -r /ray/ci/../python/requirements/ml/rllib-test-requirements.txt + # supersuit tokenizers==0.15.2 # via transformers toml==0.10.2 @@ -1929,10 +2233,13 @@ toml==0.10.2 tomli==2.0.1 # via semgrep tomlkit==0.12.3 - # via yq + # via + # snowflake-connector-python + # yq toolz==0.12.0 # via # altair + # chex # dask # distributed # partd @@ -1945,9 +2252,13 @@ torch==2.3.0 # fairscale # higher # linear-operator + # monai + # mosaicml # pyro-ppl # pytorch-lightning + # pytorch-ranger # timm + # torch-optimizer # torchmetrics # torchtext # torchvision @@ -1955,6 +2266,8 @@ torch-cluster==1.6.3 # via -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt torch-geometric==2.5.3 # via -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt +torch-optimizer==0.3.0 + # via mosaicml torch-scatter==2.1.2 # via -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt torch-sparse==0.6.18 @@ -1964,12 +2277,14 @@ torch-spline-conv==1.2.2 torchmetrics==0.10.3 # via # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt + # mosaicml # pytorch-lightning torchtext==0.18.0 # via -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt torchvision==0.18.0 # via # -r /ray/ci/../python/requirements/ml/dl-cpu-requirements.txt + # mosaicml # timm tornado==6.1 # via @@ -1979,21 +2294,27 @@ tornado==6.1 # jupyter-client # jupyter-server # jupyterlab + # nbclassic # notebook + # pymars # terminado tqdm==4.64.1 # via # -r /ray/ci/../python/requirements/test-requirements.txt + # aim # cmdstanpy # datasets # deepspeed + # dm-control # etils # evaluate # huggingface-hub # hyperopt + # mosaicml # moviepy # optuna # prophet + # pymars # pyro-ppl # pytorch-lightning # statsforecast @@ -2012,37 +2333,44 @@ traitlets==5.14.0 # jupyter-events # jupyter-server # matplotlib-inline + # nbclassic # nbclient # nbconvert # nbformat # notebook transformers==4.36.2 - # via -r /ray/ci/../python/requirements/ml/core-requirements.txt -triad==0.9.6 + # via + # -r /ray/ci/../python/requirements/ml/core-requirements.txt + # mosaicml +triad==0.9.3 # via # adagio # fugue + # fugue-sql-antlr + # qpd trustme==0.9.0 # via -r /ray/ci/../python/requirements/test-requirements.txt + # via -r /ray/ci/../python/requirements/ml/tune-requirements.txt typeguard==2.13.3 # via ax-platform typer==0.9.0 # via -r /ray/ci/../python/requirements.txt -types-python-dateutil==2.9.0.20240316 +types-python-dateutil==2.8.19.14 # via arrow types-pyyaml==6.0.12.2 # via -r /ray/ci/../python/requirements/lint-requirements.txt typing-extensions==4.8.0 # via # -r /ray/ci/../python/requirements/test-requirements.txt - # aiosqlite # alembic # aws-sam-translator # azure-core + # azure-identity # bokeh # configspace # etils # fastapi + # flax # gradio # gradio-client # gymnasium @@ -2051,16 +2379,15 @@ typing-extensions==4.8.0 # mypy # myst-nb # nevergrad - # optree + # orbax-checkpoint # pydantic # pydantic-core # pytorch-lightning # semgrep + # snowflake-connector-python # tensorflow # torch # typer -tzdata==2024.1 - # via pandas ujson==5.8.0 # via python-lsp-jsonrpc uri-template==1.3.0 @@ -2079,13 +2406,16 @@ urllib3==1.26.18 # responses # semgrep # sentry-sdk -utilsforecast==0.1.10 +utilsforecast==0.0.23 # via statsforecast uvicorn==0.22.0 # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/test-requirements.txt + # aim # gradio +uvloop==0.19.0 + # via pymars # via # nsx-policy-python-sdk # nsx-python-sdk @@ -2109,6 +2439,7 @@ virtualenv==20.25.3 # via # -r /ray/ci/../python/requirements.txt # -r /ray/ci/../python/requirements/test-requirements.txt + # pytest-virtualenv # via vsphere-automation-sdk # via vsphere-automation-sdk vsphere-automation-sdk @ git+https://github.com/vmware/vsphere-automation-sdk-python.git@v8.0.1.0 @@ -2123,7 +2454,7 @@ wcmatch==8.5 # via semgrep wcwidth==0.2.12 # via prompt-toolkit -webcolors==24.6.0 +webcolors==1.13 # via jsonschema webdataset==0.2.86 # via -r /ray/ci/../python/requirements/ml/data-test-requirements.txt @@ -2162,7 +2493,7 @@ wrapt==1.14.1 # comet-ml # tensorflow # tensorflow-datasets -wurlitzer==3.1.0 +wurlitzer==3.0.3 # via comet-ml xgboost==1.7.6 # via -r /ray/ci/../python/requirements/ml/core-requirements.txt @@ -2171,6 +2502,7 @@ xlrd==2.0.1 xmltodict==0.13.0 # via # moto + # py3nvml # yq xxhash==3.4.1 # via @@ -2180,6 +2512,8 @@ y-py==0.6.2 # via # jupyter-ydoc # ypy-websocket +yahp==0.1.1 + # via mosaicml yarl==1.9.3 # via aiohttp ypy-websocket==0.8.4