Skip to content

Commit 72c3687

Browse files
qinyiyanqinyiyan
authored
Merge pull request #28 from Hilly12/uv-setup
Setup package with uv and update requirements.
2 parents c43d6ae + 6671881 commit 72c3687

File tree

5 files changed

+2156
-19
lines changed

5 files changed

+2156
-19
lines changed

‎.gitignore

+174-2
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,174 @@
1-
__pycache__/*
2-
.ipynb_checkpoints/*
1+
# Byte-compiled / optimized / DLL files
2+
__pycache__/
3+
*.py[cod]
4+
*$py.class
5+
6+
# C extensions
7+
*.so
8+
9+
# Distribution / packaging
10+
.Python
11+
build/
12+
develop-eggs/
13+
dist/
14+
downloads/
15+
eggs/
16+
.eggs/
17+
lib/
18+
lib64/
19+
parts/
20+
sdist/
21+
var/
22+
wheels/
23+
share/python-wheels/
24+
*.egg-info/
25+
.installed.cfg
26+
*.egg
27+
MANIFEST
28+
29+
# PyInstaller
30+
# Usually these files are written by a python script from a template
31+
# before PyInstaller builds the exe, so as to inject date/other infos into it.
32+
*.manifest
33+
*.spec
34+
35+
# Installer logs
36+
pip-log.txt
37+
pip-delete-this-directory.txt
38+
39+
# Unit test / coverage reports
40+
htmlcov/
41+
.tox/
42+
.nox/
43+
.coverage
44+
.coverage.*
45+
.cache
46+
nosetests.xml
47+
coverage.xml
48+
*.cover
49+
*.py,cover
50+
.hypothesis/
51+
.pytest_cache/
52+
cover/
53+
54+
# Translations
55+
*.mo
56+
*.pot
57+
58+
# Django stuff:
59+
*.log
60+
local_settings.py
61+
db.sqlite3
62+
db.sqlite3-journal
63+
64+
# Flask stuff:
65+
instance/
66+
.webassets-cache
67+
68+
# Scrapy stuff:
69+
.scrapy
70+
71+
# Sphinx documentation
72+
docs/_build/
73+
74+
# PyBuilder
75+
.pybuilder/
76+
target/
77+
78+
# Jupyter Notebook
79+
.ipynb_checkpoints
80+
81+
# IPython
82+
profile_default/
83+
ipython_config.py
84+
85+
# pyenv
86+
# For a library or package, you might want to ignore these files since the code is
87+
# intended to run in multiple environments; otherwise, check them in:
88+
# .python-version
89+
90+
# pipenv
91+
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92+
# However, in case of collaboration, if having platform-specific dependencies or dependencies
93+
# having no cross-platform support, pipenv may install dependencies that don't work, or not
94+
# install all needed dependencies.
95+
#Pipfile.lock
96+
97+
# UV
98+
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
99+
# This is especially recommended for binary packages to ensure reproducibility, and is more
100+
# commonly ignored for libraries.
101+
#uv.lock
102+
103+
# poetry
104+
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
105+
# This is especially recommended for binary packages to ensure reproducibility, and is more
106+
# commonly ignored for libraries.
107+
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
108+
#poetry.lock
109+
110+
# pdm
111+
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
112+
#pdm.lock
113+
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
114+
# in version control.
115+
# https://pdm.fming.dev/latest/usage/project/#working-with-version-control
116+
.pdm.toml
117+
.pdm-python
118+
.pdm-build/
119+
120+
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
121+
__pypackages__/
122+
123+
# Celery stuff
124+
celerybeat-schedule
125+
celerybeat.pid
126+
127+
# SageMath parsed files
128+
*.sage.py
129+
130+
# Environments
131+
.env
132+
.venv
133+
env/
134+
venv/
135+
ENV/
136+
env.bak/
137+
venv.bak/
138+
139+
# Spyder project settings
140+
.spyderproject
141+
.spyproject
142+
143+
# Rope project settings
144+
.ropeproject
145+
146+
# mkdocs documentation
147+
/site
148+
149+
# mypy
150+
.mypy_cache/
151+
.dmypy.json
152+
dmypy.json
153+
154+
# Pyre type checker
155+
.pyre/
156+
157+
# pytype static type analyzer
158+
.pytype/
159+
160+
# Cython debug symbols
161+
cython_debug/
162+
163+
# PyCharm
164+
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
165+
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
166+
# and can be added to the global gitignore or merged into this file. For a more nuclear
167+
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
168+
#.idea/
169+
170+
# Ruff stuff:
171+
.ruff_cache/
172+
173+
# PyPI configuration file
174+
.pypirc

‎.python-version

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
3.12

‎pyproject.toml

+33
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
[project]
2+
name = "recml"
3+
version = "0.1.1"
4+
description = "RecML is envisioned as a high-performance, large-scale deep learning recommender system library optimized for Cloud TPUs"
5+
readme = "README.md"
6+
requires-python = ">=3.12"
7+
license-files = ["LICENSE"]
8+
dynamic = ["dependencies"]
9+
10+
[dependency-groups]
11+
dev = [
12+
"einops>=0.8.1",
13+
"ml-dtypes>=0.5.1",
14+
"pre-commit>=4.2.0",
15+
"pylint>=3.3.6",
16+
"pytest>=8.3.5",
17+
"pytest-env>=1.1.5",
18+
"torch>=2.6.0",
19+
]
20+
jax = [
21+
"clu>=0.0.12",
22+
"fiddle>=0.3.0",
23+
"jax>=0.6.0",
24+
"jaxtyping>=0.3.1",
25+
"keras>=3.9.2",
26+
"keras-hub>=0.20.0",
27+
"tensorflow>=2.19.0",
28+
"tensorflow-datasets>=4.9.8",
29+
]
30+
31+
32+
[tool.uv]
33+
default-groups = ["all"]

‎requirements.txt

+104-17
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,104 @@
1-
jax>=0.4.30
2-
jaxlib>=0.4.30
3-
numpy
4-
pylint
5-
pytest
6-
pyink
7-
pre-commit
8-
clu
9-
jaxtyping
10-
tensorflow
11-
keras
12-
fiddle
13-
tensorflow_datasets
14-
absl-py
15-
keras_hub
16-
torch
17-
einops
1+
absl-py==2.2.2
2+
astroid==3.3.9
3+
astunparse==1.6.3
4+
attrs==25.3.0
5+
certifi==2025.1.31
6+
cfgv==3.4.0
7+
charset-normalizer==3.4.1
8+
chex==0.1.89
9+
clu==0.0.12
10+
dill==0.4.0
11+
distlib==0.3.9
12+
dm-tree==0.1.9
13+
docstring-parser==0.16
14+
einops==0.8.1
15+
etils==1.12.2
16+
fiddle==0.3.0
17+
filelock==3.18.0
18+
flatbuffers==25.2.10
19+
flax==0.10.5
20+
fsspec==2025.3.2
21+
gast==0.6.0
22+
google-pasta==0.2.0
23+
googleapis-common-protos==1.70.0
24+
graphviz==0.20.3
25+
grpcio==1.71.0
26+
h5py==3.13.0
27+
humanize==4.12.2
28+
identify==2.6.9
29+
idna==3.10
30+
immutabledict==4.2.1
31+
importlib-resources==6.5.2
32+
iniconfig==2.1.0
33+
isort==6.0.1
34+
jax==0.6.0
35+
jaxlib==0.6.0
36+
jaxtyping==0.3.1
37+
jinja2==3.1.6
38+
kagglehub==0.3.11
39+
keras==3.9.2
40+
keras-hub==0.20.0
41+
libclang==18.1.1
42+
libcst==1.7.0
43+
markdown==3.8
44+
markdown-it-py==3.0.0
45+
markupsafe==3.0.2
46+
mccabe==0.7.0
47+
mdurl==0.1.2
48+
ml-collections==1.1.0
49+
ml-dtypes==0.5.1
50+
mpmath==1.3.0
51+
msgpack==1.1.0
52+
namex==0.0.8
53+
nest-asyncio==1.6.0
54+
networkx==3.4.2
55+
nodeenv==1.9.1
56+
numpy==2.1.3
57+
opt-einsum==3.4.0
58+
optax==0.2.4
59+
optree==0.15.0
60+
orbax-checkpoint==0.11.12
61+
packaging==24.2
62+
platformdirs==4.3.7
63+
pluggy==1.5.0
64+
pre-commit==4.2.0
65+
promise==2.3
66+
protobuf==5.29.4
67+
psutil==7.0.0
68+
pyarrow==19.0.1
69+
pygments==2.19.1
70+
pylint==3.3.6
71+
pytest==8.3.5
72+
pytest-env==1.1.5
73+
pyyaml==6.0.2
74+
regex==2024.11.6
75+
requests==2.32.3
76+
rich==14.0.0
77+
scipy==1.15.2
78+
setuptools==78.1.0
79+
simple-parsing==0.1.7
80+
simplejson==3.20.1
81+
six==1.17.0
82+
sympy==1.13.1
83+
tensorboard==2.19.0
84+
tensorboard-data-server==0.7.2
85+
tensorflow==2.19.0
86+
tensorflow-datasets==4.9.8
87+
tensorflow-metadata==1.17.1
88+
tensorflow-text==2.19.0
89+
tensorstore==0.1.73
90+
termcolor==3.0.1
91+
toml==0.10.2
92+
tomlkit==0.13.2
93+
toolz==1.0.0
94+
torch==2.6.0
95+
tqdm==4.67.1
96+
treescope==0.1.9
97+
typing-extensions==4.13.2
98+
urllib3==2.4.0
99+
virtualenv==20.30.0
100+
wadler-lindig==0.1.5
101+
werkzeug==3.1.3
102+
wheel==0.45.1
103+
wrapt==1.17.2
104+
zipp==3.21.0

0 commit comments

Comments
 (0)