flwr 1.23.0__py3-none-any.whl → 1.25.0__py3-none-any.whl

flwr/cli/new/templates/app/code/task.sklearn.py.tpl

@@ -1,67 +0,0 @@
-"""$project_name: A Flower / $framework_str app."""
-
-import numpy as np
-from flwr_datasets import FederatedDataset
-from flwr_datasets.partitioner import IidPartitioner
-from sklearn.linear_model import LogisticRegression
-
-fds = None  # Cache FederatedDataset
-
-
-def load_data(partition_id: int, num_partitions: int):
-    """Load partition MNIST data."""
-    # Only initialize `FederatedDataset` once
-    global fds
-    if fds is None:
-        partitioner = IidPartitioner(num_partitions=num_partitions)
-        fds = FederatedDataset(
-            dataset="ylecun/mnist",
-            partitioners={"train": partitioner},
-        )
-
-    dataset = fds.load_partition(partition_id, "train").with_format("numpy")
-
-    X, y = dataset["image"].reshape((len(dataset), -1)), dataset["label"]
-
-    # Split the on edge data: 80% train, 20% test
-    X_train, X_test = X[: int(0.8 * len(X))], X[int(0.8 * len(X)) :]
-    y_train, y_test = y[: int(0.8 * len(y))], y[int(0.8 * len(y)) :]
-
-    return X_train, X_test, y_train, y_test
-
-
-def get_model(penalty: str, local_epochs: int):
-
-    return LogisticRegression(
-        penalty=penalty,
-        max_iter=local_epochs,
-        warm_start=True,
-    )
-
-
-def get_model_params(model):
-    if model.fit_intercept:
-        params = [
-            model.coef_,
-            model.intercept_,
-        ]
-    else:
-        params = [model.coef_]
-    return params
-
-
-def set_model_params(model, params):
-    model.coef_ = params[0]
-    if model.fit_intercept:
-        model.intercept_ = params[1]
-    return model
-
-
-def set_initial_params(model):
-    n_classes = 10  # MNIST has 10 classes
-    n_features = 784  # Number of features in dataset
-    model.classes_ = np.array([i for i in range(10)])
-
-    model.coef_ = np.zeros((n_classes, n_features))
-    if model.fit_intercept:
-        model.intercept_ = np.zeros((n_classes,))

flwr/cli/new/templates/app/code/task.tensorflow.py.tpl

@@ -1,52 +0,0 @@
-"""$project_name: A Flower / $framework_str app."""
-
-import os
-
-import keras
-from flwr_datasets import FederatedDataset
-from flwr_datasets.partitioner import IidPartitioner
-from keras import layers
-
-# Make TensorFlow log less verbose
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
-
-
-def load_model():
-    # Define a simple CNN for CIFAR-10 and set Adam optimizer
-    model = keras.Sequential(
-        [
-            keras.Input(shape=(32, 32, 3)),
-            layers.Conv2D(32, kernel_size=(3, 3), activation="relu"),
-            layers.MaxPooling2D(pool_size=(2, 2)),
-            layers.Conv2D(64, kernel_size=(3, 3), activation="relu"),
-            layers.MaxPooling2D(pool_size=(2, 2)),
-            layers.Flatten(),
-            layers.Dropout(0.5),
-            layers.Dense(10, activation="softmax"),
-        ]
-    )
-    model.compile("adam", "sparse_categorical_crossentropy", metrics=["accuracy"])
-    return model
-
-
-fds = None  # Cache FederatedDataset
-
-
-def load_data(partition_id, num_partitions):
-    # Download and partition dataset
-    # Only initialize `FederatedDataset` once
-    global fds
-    if fds is None:
-        partitioner = IidPartitioner(num_partitions=num_partitions)
-        fds = FederatedDataset(
-            dataset="uoft-cs/cifar10",
-            partitioners={"train": partitioner},
-        )
-    partition = fds.load_partition(partition_id, "train")
-    partition.set_format("numpy")
-
-    # Divide data on each node: 80% train, 20% test
-    partition = partition.train_test_split(test_size=0.2)
-    x_train, y_train = partition["train"]["img"] / 255.0, partition["train"]["label"]
-    x_test, y_test = partition["test"]["img"] / 255.0, partition["test"]["label"]
-    return x_train, y_train, x_test, y_test

flwr/cli/new/templates/app/code/task.xgboost.py.tpl

@@ -1,67 +0,0 @@
-"""$project_name: A Flower / $framework_str app."""
-
-import xgboost as xgb
-from flwr_datasets import FederatedDataset
-from flwr_datasets.partitioner import IidPartitioner
-
-
-def train_test_split(partition, test_fraction, seed):
-    """Split the data into train and validation set given split rate."""
-    train_test = partition.train_test_split(test_size=test_fraction, seed=seed)
-    partition_train = train_test["train"]
-    partition_test = train_test["test"]
-
-    num_train = len(partition_train)
-    num_test = len(partition_test)
-
-    return partition_train, partition_test, num_train, num_test
-
-
-def transform_dataset_to_dmatrix(data):
-    """Transform dataset to DMatrix format for xgboost."""
-    x = data["inputs"]
-    y = data["label"]
-    new_data = xgb.DMatrix(x, label=y)
-    return new_data
-
-
-fds = None  # Cache FederatedDataset
-
-
-def load_data(partition_id, num_clients):
-    """Load partition HIGGS data."""
-    # Only initialize `FederatedDataset` once
-    global fds
-    if fds is None:
-        partitioner = IidPartitioner(num_partitions=num_clients)
-        fds = FederatedDataset(
-            dataset="jxie/higgs",
-            partitioners={"train": partitioner},
-        )
-
-    # Load the partition for this `partition_id`
-    partition = fds.load_partition(partition_id, split="train")
-    partition.set_format("numpy")
-
-    # Train/test splitting
-    train_data, valid_data, num_train, num_val = train_test_split(
-        partition, test_fraction=0.2, seed=42
-    )
-
-    # Reformat data to DMatrix for xgboost
-    train_dmatrix = transform_dataset_to_dmatrix(train_data)
-    valid_dmatrix = transform_dataset_to_dmatrix(valid_data)
-
-    return train_dmatrix, valid_dmatrix, num_train, num_val
-
-
-def replace_keys(input_dict, match="-", target="_"):
-    """Recursively replace match string with target string in dictionary keys."""
-    new_dict = {}
-    for key, value in input_dict.items():
-        new_key = key.replace(match, target)
-        if isinstance(value, dict):
-            new_dict[new_key] = replace_keys(value, match, target)
-        else:
-            new_dict[new_key] = value
-    return new_dict

flwr/cli/new/templates/app/code/utils.baseline.py.tpl

@@ -1 +0,0 @@
-"""$project_name: A Flower Baseline."""

flwr/cli/new/templates/app/pyproject.baseline.toml.tpl

@@ -1,146 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "flwr-datasets[vision]>=0.5.0",
-    "torch==2.8.0",
-    "torchvision==0.23.0",
-]
-
-[tool.hatch.metadata]
-allow-direct-references = true
-
-[project.optional-dependencies]
-dev = [
-    "isort==5.13.2",
-    "black==24.2.0",
-    "docformatter==1.7.5",
-    "mypy==1.8.0",
-    "pylint==3.3.1",
-    "pytest==7.4.4",
-    "pytest-watch==4.2.0",
-    "ruff==0.4.5",
-    "types-requests==2.31.0.20240125",
-]
-
-[tool.isort]
-profile = "black"
-
-[tool.black]
-line-length = 88
-target-version = ["py310", "py311", "py312"]
-
-[tool.pytest.ini_options]
-minversion = "6.2"
-addopts = "-qq"
-
-[tool.mypy]
-ignore_missing_imports = true
-strict = false
-plugins = "numpy.typing.mypy_plugin"
-
-[tool.pylint."MESSAGES CONTROL"]
-disable = "duplicate-code,too-few-public-methods,useless-import-alias"
-good-names = "i,j,k,_,x,y,X,Y,K,N"
-max-args = 10
-max-attributes = 15
-max-locals = 36
-max-branches = 20
-max-statements = 55
-
-[tool.pylint.typecheck]
-generated-members = "numpy.*, torch.*, tensorflow.*"
-
-[[tool.mypy.overrides]]
-module = [
-    "importlib.metadata.*",
-    "importlib_metadata.*",
-]
-follow_imports = "skip"
-follow_imports_for_stubs = true
-disallow_untyped_calls = false
-
-[[tool.mypy.overrides]]
-module = "torch.*"
-follow_imports = "skip"
-follow_imports_for_stubs = true
-
-[tool.docformatter]
-wrap-summaries = 88
-wrap-descriptions = 88
-
-[tool.ruff]
-target-version = "py310"
-line-length = 88
-exclude = [
-    ".bzr",
-    ".direnv",
-    ".eggs",
-    ".git",
-    ".hg",
-    ".mypy_cache",
-    ".nox",
-    ".pants.d",
-    ".pytype",
-    ".ruff_cache",
-    ".svn",
-    ".tox",
-    ".venv",
-    "__pypackages__",
-    "_build",
-    "buck-out",
-    "build",
-    "dist",
-    "node_modules",
-    "venv",
-    "proto",
-]
-
-[tool.ruff.lint]
-select = ["D", "E", "F", "W", "B", "ISC", "C4", "UP"]
-fixable = ["D", "E", "F", "W", "B", "ISC", "C4", "UP"]
-ignore = ["B024", "B027", "D205", "D209"]
-
-[tool.ruff.lint.pydocstyle]
-convention = "numpy"
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-# Format: "<module>:<object>"
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-num-server-rounds = 3
-fraction-train = 0.5
-local-epochs = 1
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "local-simulation"
-
-# Local simulation federation with 10 virtual SuperNodes
-[tool.flwr.federations.local-simulation]
-options.num-supernodes = 10
-options.backend.client-resources.num-cpus = 2
-options.backend.client-resources.num-gpus = 0.0

flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl

@@ -1,80 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "flwr-datasets>=0.5.0",
-    "torch==2.4.0",
-    "trl==0.8.1",
-    "bitsandbytes==0.45.4",
-    "scipy==1.13.0",
-    "peft==0.6.2",
-    "transformers==4.50.3",
-    "sentencepiece==0.2.0",
-    "omegaconf==2.3.0",
-    "hf_transfer==0.1.8",
-]
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-# Format: "<module>:<object>"
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-model.name = "mistralai/Mistral-7B-v0.3"
-model.quantization = 4
-model.gradient-checkpointing = true
-model.lora.peft-lora-r = 32
-model.lora.peft-lora-alpha = 64
-train.save-every-round = 5
-train.learning-rate-max = 5e-5
-train.learning-rate-min = 1e-6
-train.seq-length = 512
-train.training-arguments.output-dir = ""
-train.training-arguments.learning-rate = ""
-train.training-arguments.per-device-train-batch-size = 16
-train.training-arguments.gradient-accumulation-steps = 1
-train.training-arguments.logging-steps = 10
-train.training-arguments.num-train-epochs = 3
-train.training-arguments.max-steps = 10
-train.training-arguments.save-steps = 1000
-train.training-arguments.save-total-limit = 10
-train.training-arguments.gradient-checkpointing = true
-train.training-arguments.lr-scheduler-type = "constant"
-strategy.fraction-train = $fraction_train
-strategy.fraction-evaluate = 0.0
-num-server-rounds = 200
-
-# Dataset config (static for FlowerTune LLM Leaderboard)
-[tool.flwr.app.config.static]
-dataset.name = "$dataset_name"
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "local-simulation"
-
-# Local simulation federation with $num_clients virtual SuperNodes
-[tool.flwr.federations.local-simulation]
-options.num-supernodes = $num_clients
-options.backend.client-resources.num-cpus = 6
-options.backend.client-resources.num-gpus = 1.0

flwr/cli/new/templates/app/pyproject.huggingface.toml.tpl

@@ -1,65 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "flwr-datasets>=0.5.0",
-    "torch>=2.7.1",
-    "transformers>=4.30.0,<5.0",
-    "evaluate>=0.4.0,<1.0",
-    "datasets>=2.0.0, <3.0",
-    "scikit-learn>=1.6.1, <2.0",
-]
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-# Format: "<module>:<object>"
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-num-server-rounds = 3
-fraction-train = 0.5
-local-steps = 5
-model-name = "prajjwal1/bert-tiny" # Set a larger model if you have access to more GPU resources
-num-labels = 2
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "localhost"
-
-# Local simulation federation with 10 virtual SuperNodes
-[tool.flwr.federations.localhost]
-options.num-supernodes = 10
-
-# Local simulation federation with 10 virtual SuperNodes
-# making use of GPUs
-[tool.flwr.federations.localhost-gpu]
-options.num-supernodes = 10
-options.backend.client-resources.num-cpus = 4 # each ClientApp assumes to use 4CPUs
-options.backend.client-resources.num-gpus = 0.25 # at most 4 ClientApps will run in a given GPU
-
-# Remote federation example for use with SuperLink
-[tool.flwr.federations.remote-federation]
-address = "<SUPERLINK-ADDRESS>:<PORT>"
-insecure = true  # Remove this line to enable TLS
-# root-certificates = "<PATH/TO/ca.crt>"  # For TLS setup

flwr/cli/new/templates/app/pyproject.jax.toml.tpl

@@ -1,52 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "jax==0.4.30",
-    "jaxlib==0.4.30",
-    "scikit-learn==1.6.1",
-]
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-# Format: "<module>:<object>"
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-num-server-rounds = 3
-input-dim = 3
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "local-simulation"
-
-# Local simulation federation with 10 virtual SuperNodes
-[tool.flwr.federations.local-simulation]
-options.num-supernodes = 10
-
-# Remote federation example for use with SuperLink
-[tool.flwr.federations.remote-federation]
-address = "<SUPERLINK-ADDRESS>:<PORT>"
-insecure = true  # Remove this line to enable TLS
-# root-certificates = "<PATH/TO/ca.crt>"  # For TLS setup

flwr/cli/new/templates/app/pyproject.mlx.toml.tpl

@@ -1,56 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "flwr-datasets[vision]>=0.5.0",
-    "mlx==0.29.0",
-]
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-# Format: "<module>:<object>"
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-num-server-rounds = 3
-local-epochs = 1
-num-layers = 2
-input-dim = 784 # 28*28
-hidden-dim = 32
-batch-size = 256
-lr = 0.1
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "local-simulation"
-
-# Local simulation federation with 10 virtual SuperNodes
-[tool.flwr.federations.local-simulation]
-options.num-supernodes = 10
-
-# Remote federation example for use with SuperLink
-[tool.flwr.federations.remote-federation]
-address = "<SUPERLINK-ADDRESS>:<PORT>"
-insecure = true  # Remove this line to enable TLS
-# root-certificates = "<PATH/TO/ca.crt>"  # For TLS setup

flwr/cli/new/templates/app/pyproject.numpy.toml.tpl

@@ -1,49 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "numpy>=2.0.2",
-]
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-# Format: "<module>:<object>"
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-num-server-rounds = 3
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "local-simulation"
-
-# Local simulation federation with 10 virtual SuperNodes
-[tool.flwr.federations.local-simulation]
-options.num-supernodes = 10
-
-# Remote federation example for use with SuperLink
-[tool.flwr.federations.remote-federation]
-address = "<SUPERLINK-ADDRESS>:<PORT>"
-insecure = true  # Remove this line to enable TLS
-# root-certificates = "<PATH/TO/ca.crt>"  # For TLS setup

flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl

@@ -1,53 +0,0 @@
-# =====================================================================
-# For a full TOML configuration guide, check the Flower docs:
-# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
-# =====================================================================
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[project]
-name = "$package_name"
-version = "1.0.0"
-description = ""
-license = "Apache-2.0"
-# Dependencies for your Flower App
-dependencies = [
-    "flwr[simulation]>=1.23.0",
-    "flwr-datasets[vision]>=0.5.0",
-    "torch==2.7.1",
-    "torchvision==0.22.1",
-]
-
-[tool.hatch.build.targets.wheel]
-packages = ["."]
-
-[tool.flwr.app]
-publisher = "$username"
-
-# Point to your ServerApp and ClientApp objects
-[tool.flwr.app.components]
-serverapp = "$import_name.server_app:app"
-clientapp = "$import_name.client_app:app"
-
-# Custom config values accessible via `context.run_config`
-[tool.flwr.app.config]
-num-server-rounds = 3
-fraction-train = 0.5
-local-epochs = 1
-lr = 0.01
-
-# Default federation to use when running the app
-[tool.flwr.federations]
-default = "local-simulation"
-
-# Local simulation federation with 10 virtual SuperNodes
-[tool.flwr.federations.local-simulation]
-options.num-supernodes = 10
-
-# Remote federation example for use with SuperLink
-[tool.flwr.federations.remote-federation]
-address = "<SUPERLINK-ADDRESS>:<PORT>"
-insecure = true  # Remove this line to enable TLS
-# root-certificates = "<PATH/TO/ca.crt>"  # For TLS setup