Skip to content

Commit

Permalink
adding runtimes and pinning versions
Browse files Browse the repository at this point in the history
  • Loading branch information
jeandut committed Nov 18, 2024
1 parent da73436 commit a081126
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 12 deletions.
10 changes: 9 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,16 @@ To install the package, create an env with python `3.9` with conda
conda create -n fedeca python=3.9
conda activate fedeca
```

This should take under 20 seconds with a reasonably good connexion.
Within the environment, install the package by running:

```
git clone https://github.com/owkin/fedeca.git
cd fedeca
pip install -e ".[all_extra]"
```
The cloning should take less than 5 seconds and the pip install
should take around 1 min and 30 seconds.

If you plan on developing, you should also install the pre-commit hooks

Expand All @@ -40,10 +42,16 @@ pre-commit install

This will run all the pre-commit hooks at each commit, ensuring a clean repo.

All estimated runtimes computations use as reference machine a Mac M3 Pro with 18G of RAM
and an internet connexion at 120Mbps as measured by fast.com.

## Quickstart

Go [here](./quickstart/quickstart.md).

Copy-pasting all cells of the tutorial into a regular python script gives a runtime of approximately 1 minute,
with less than 2 seconds spent on each run of the FedECA algorithm in-RAM.

## <a name="citation"></a>Citing FedECA

```
Expand Down
24 changes: 13 additions & 11 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,22 +31,24 @@
install_requires=[
# We cannot use the git+https / git+ssh syntax here because of docker
# build issues related to git not being installed
# Commented pinned versions are only there as references but should
# not matter as long as the versions are compatible with the ones
# uncommented.
"substrafl==0.46.0",
"argparse",
"numpy==1.26.4",
"pandas",
"pre-commit",
"scipy",
"seaborn",
"opacus",
"lifelines",
"git-python",
"build",
"pandas",#==2.2.3
"pre-commit",#==4.0.1
"scipy",#==1.13.1
"seaborn",#==0.13.2
"opacus",#==1.4.0
"lifelines",#==0.29.0
"git-python",#==1.0.3
"build",#==1.2.2.post1
"torch==1.13.1",
"scikit-learn==1.2.1",
"pydantic", # Need to be updated to > 2.0 to use latest Substra
"pydantic", #==2.23.4
"indcomp==0.2.1",
"hydra-core",
"hydra-core",#==1.3.2
],
extras_require={
"all_extra": all_extra,
Expand Down

0 comments on commit a081126

Please sign in to comment.