google-research / google-research/big_vision

Add a pyproject.toml

Open
#150 0 comments 2 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
3.5k
Forks
227
PR merge metrics
No merged PRs in 30d

Description

Hi,

I found it non-trivial to set up a vanilla python environment to work with `big_vision`. Would including a pyproject.toml in the root be valuable for the project? If so I could open a PR.

Here's the config that works for me:

```toml
[project]
name = "big-vision"
version = "0.1.0"
requires-python = ">=3.10,<3.11"
dependencies = [
"absl-py>=2.1.0",
"clu",
"distrax>=0.1.5",
"editdistance>=0.8.1",
"einops>=0.8.0",
"flax>=0.10.2",
"flaxformer",
"numpy<2.0.0",
"optax>=0.2.4",
"overrides>=7.7.0",
"protobuf<3.21.0",
"psutil>=6.1.1",
"pycocoevalcap>=1.2",
"sentencepiece>=0.2.0",
"tensorflow-cpu>=2.10.0",
"tensorflow-gan>=2.1.0",
"tensorflow-text>=2.9.0",
"tfds-nightly>=4.9.0.dev202304110044",
"keras<3.0.0",
"jax[cuda]>=0.4.34",
]

[tool.uv.sources]
flaxformer = { git = "https://github.com/google/flaxformer" }
clu = { git = "https://github.com/google/CommonLoopUtils" }
```

Contributor guide

Open the contributing guide

Research direction

Start by reviewing the proposed root pyproject.toml configuration and its dependency constraints from the issue. Confirm that it provides a usable vanilla Python environment for big_vision; done should include a project metadata file with the required dependencies and git sources.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, tensorflow
Domain
build-system, developer-experience
Issue type
Feature
Difficulty
2/5
Estimated time
1-3 hours
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.