diff --git a/poetry.lock b/poetry.lock index f8b07272..1a90df1a 100644 --- a/poetry.lock +++ b/poetry.lock @@ -3063,18 +3063,23 @@ name = "lightray" version = "0.1.5" description = "Distribute a LightningCLI hyperparameter search with Ray Tune" optional = false -python-versions = "<4.0,>=3.9" -files = [ - {file = "lightray-0.1.5-py3-none-any.whl", hash = "sha256:5d5b8cdb54e3990d95bfd8147b86f7651cc632685514db9150a069994252a508"}, - {file = "lightray-0.1.5.tar.gz", hash = "sha256:6677facd2cbb8292bde00efcf1aec13ff6342598503900c4fa4f85a54f1473e1"}, -] +python-versions = "^3.9" +files = [] +develop = false [package.dependencies] -jsonargparse = {version = ">=4.29,<5.0", extras = ["signatures"]} +jsonargparse = {version = "^4.33.2", extras = ["signatures"]} lightning = "<=2.4.0" -pyarrow = ">=17.0.0,<18.0.0" -ray = {version = ">=2.8.0,<3.0.0", extras = ["default", "tune"]} -s3fs = ">=2024.9.0,<2025.0.0" +pyarrow = "^17.0.0" +ray = {version = "^2.8.0", extras = ["default", "tune"]} +s3fs = "^2024.9.0" +wandb = "^0.18.5" + +[package.source] +type = "git" +url = "https://github.com/EthanMarx/lightray.git" +reference = "refactor" +resolved_reference = "00b4d6d012ab0cc9c61330ca0be2edf63c12db35" [[package]] name = "ligo-segments" @@ -6902,45 +6907,52 @@ test = ["covdefaults (>=2.3)", "coverage (>=7.2.7)", "coverage-enable-subprocess [[package]] name = "wandb" -version = "0.16.6" +version = "0.18.5" description = "A CLI and library for interacting with the Weights & Biases API." optional = false python-versions = ">=3.7" files = [ - {file = "wandb-0.16.6-py3-none-any.whl", hash = "sha256:5810019a3b981c796e98ea58557a7c380f18834e0c6bdaed15df115522e5616e"}, - {file = "wandb-0.16.6.tar.gz", hash = "sha256:86f491e3012d715e0d7d7421a4d6de41abef643b7403046261f962f3e512fe1c"}, + {file = "wandb-0.18.5-py3-none-any.whl", hash = "sha256:49ba7bafff0cecff2159bc6fb68176d6e5561d744a9bd6a63753e7077a74e26d"}, + {file = "wandb-0.18.5-py3-none-macosx_10_13_x86_64.whl", hash = "sha256:c9d903dbff9517843881d9a0d561d82bcf0e949d8b8c03aafa35aceef31ea7e0"}, + {file = "wandb-0.18.5-py3-none-macosx_11_0_arm64.whl", hash = "sha256:33d3e5765a9bb305558af4f291338cf8723856d2b3a3c377414cd8f8b711baa4"}, + {file = "wandb-0.18.5-py3-none-macosx_11_0_x86_64.whl", hash = "sha256:9ff72e7a45e998e2a7ff42645ec76bedabf17ea51fd112ae2837dce5023ac0cc"}, + {file = "wandb-0.18.5-py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:136a79c06c114c225add8f977acec23d53df5e2a4d2803700a5eb5501ae40160"}, + {file = "wandb-0.18.5-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:350fc0f6f5bc23f4baeef12ccfea6daa89d1a5b84948d72d48249cbb652ab22a"}, + {file = "wandb-0.18.5-py3-none-musllinux_1_2_x86_64.whl", hash = "sha256:4e31741eefa2b2a83aa9f69ef27ce6112fee6f4c792a9e2996cf374a74465276"}, + {file = "wandb-0.18.5-py3-none-win32.whl", hash = "sha256:b2a25e9caf63c12e5de1cbdb30b13f76553c04754f9ff404080f70386adb8384"}, + {file = "wandb-0.18.5-py3-none-win_amd64.whl", hash = "sha256:83b619167eb2ffdd1188cba3805ccad158f6fd7fc06bef43daf6d2729a787fa0"}, + {file = "wandb-0.18.5.tar.gz", hash = "sha256:75ef47ba7fc709b787be05e558f1635d99246afeacc9369031c1be6e5b620ce6"}, ] [package.dependencies] -appdirs = ">=1.4.3" -Click = ">=7.1,<8.0.0 || >8.0.0" +click = ">=7.1,<8.0.0 || >8.0.0" docker-pycreds = ">=0.4.0" -GitPython = ">=1.0.0,<3.1.29 || >3.1.29" +gitpython = ">=1.0.0,<3.1.29 || >3.1.29" +platformdirs = "*" protobuf = [ - {version = ">=3.15.0,<4.21.0 || >4.21.0,<5", markers = "python_version == \"3.9\" and sys_platform == \"linux\""}, - {version = ">=3.19.0,<4.21.0 || >4.21.0,<5", markers = "python_version > \"3.9\" or sys_platform != \"linux\""}, + {version = ">=3.15.0,<4.21.0 || >4.21.0,<5.28.0 || >5.28.0,<6", markers = "python_version == \"3.9\" and sys_platform == \"linux\""}, + {version = ">=3.19.0,<4.21.0 || >4.21.0,<5.28.0 || >5.28.0,<6", markers = "python_version > \"3.9\" or sys_platform != \"linux\""}, ] psutil = ">=5.0.0" -PyYAML = "*" +pyyaml = "*" requests = ">=2.0.0,<3" -sentry-sdk = ">=1.0.0" +sentry-sdk = ">=2.0.0" setproctitle = "*" setuptools = "*" -typing-extensions = {version = "*", markers = "python_version < \"3.10\""} +typing-extensions = {version = ">=4.4,<5", markers = "python_version < \"3.12\""} [package.extras] -async = ["httpx (>=0.23.0)"] aws = ["boto3"] azure = ["azure-identity", "azure-storage-blob"] gcp = ["google-cloud-storage"] -importers = ["filelock", "mlflow", "polars", "rich", "tenacity"] +importers = ["filelock", "mlflow", "polars (<=1.2.1)", "rich", "tenacity"] kubeflow = ["google-cloud-storage", "kubernetes", "minio", "sh"] -launch = ["PyYAML (>=6.0.0)", "awscli", "azure-containerregistry", "azure-identity", "azure-storage-blob", "boto3", "botocore", "chardet", "google-auth", "google-cloud-aiplatform", "google-cloud-artifact-registry", "google-cloud-compute", "google-cloud-storage", "iso8601", "kubernetes", "kubernetes-asyncio", "nbconvert", "nbformat", "optuna", "pydantic", "tomli", "typing-extensions"] -media = ["bokeh", "moviepy", "numpy", "pillow", "plotly (>=5.18.0)", "rdkit-pypi", "soundfile"] +launch = ["awscli", "azure-containerregistry", "azure-identity", "azure-storage-blob", "boto3", "botocore", "chardet", "google-auth", "google-cloud-aiplatform", "google-cloud-artifact-registry", "google-cloud-compute", "google-cloud-storage", "iso8601", "jsonschema", "kubernetes", "kubernetes-asyncio", "nbconvert", "nbformat", "optuna", "pydantic", "pyyaml (>=6.0.0)", "tomli", "typing-extensions"] +media = ["bokeh", "imageio", "moviepy", "numpy", "pillow", "plotly (>=5.18.0)", "rdkit", "soundfile"] models = ["cloudpickle"] perf = ["orjson"] -reports = ["pydantic (>=2.0.0)"] sweeps = ["sweeps (>=0.2.0)"] +workspaces = ["wandb-workspaces"] [[package]] name = "wcwidth" @@ -7219,4 +7231,4 @@ type = ["pytest-mypy"] [metadata] lock-version = "2.0" python-versions = ">=3.9,<3.13" -content-hash = "e423c0a406948965e80f5efab343cbad5d650d24446205d9d45a16a25cc41e72" +content-hash = "bf2487dab82b94d0689bc5d6e0d3dfbf821d9c4788bb71d51081ef0facb3aa86" diff --git a/pyproject.toml b/pyproject.toml index 94d649cb..ced922f4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,14 +18,14 @@ bilby = "^2.2.3" healpy = "^1.16.6" pyro-ppl = "^1.9.0" jsonargparse = {version = "^4.27.1", extras = ["signatures"]} -wandb = "^0.16.3" urllib3 = ">=1.25.4,<1.27" omegaconf = "^2.3.0" -lightray = "^0.1.5" +lightray = {git = "https://github.com/EthanMarx/lightray.git", branch = "refactor"} law = ">=0.1.19" mldatafind = ">=0.1.0" luigi = "^3.5.1" gwpy="3.0.8" +wandb = "^0.18.5" [tool.poetry.group.dev.dependencies] pre-commit = "^3.7.0"