colink.toml

[package]
name = "unifed-flower"
version = "0.0.0"
keywords = ["federated learning", "machine learning"]
description = "Integrating Flower framework into UniFed as a CoLink protocol."
language = "python"
install_script = """
if { conda env list | grep 'colink-protocol-unifed-flower'; } >/dev/null 2>&1; then
    conda env remove -n colink-protocol-unifed-flower
fi
conda create -n colink-protocol-unifed-flower python=3.9 -y
conda activate colink-protocol-unifed-flower
pip install .
git clone https://github.com/adap/flower.git
cd flower
git checkout d1eb90f74714a9c10ddbeefb767b56be7b61303d
pip install --upgrade pip
cp ../flower.patch ./flower.patch
git apply --whitespace=fix flower.patch
pip install .
pip install torch torchvision tensorflow scikit-learn tqdm uvicorn fastapi opencv-python-headless pytest
rm -rf ~/.cache/pip
cd ..
"""
entrypoint = """
conda activate colink-protocol-unifed-flower
unifed-flower > debug.txt 2>&1
"""