From a13cbd451a20fc0d1a899e4cabc7a7cc6f94ac8f Mon Sep 17 00:00:00 2001 From: debrevitatevitae Date: Wed, 17 Jul 2024 16:41:06 +0200 Subject: [PATCH] Fix docs --- pyproject.toml | 2 +- qadence/ml_tools/train_grad.py | 4 ---- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index fbbf1207..a1368d00 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -140,7 +140,7 @@ dependencies = [ "markdown-exec", "mike", ] -features = ["pulser", "braket", "horqrux", "visualization"] +features = ["pulser", "braket", "horqrux", "visualization", "mlflow"] [tool.hatch.envs.docs.scripts] build = "mkdocs build --clean --strict" diff --git a/qadence/ml_tools/train_grad.py b/qadence/ml_tools/train_grad.py index d0426562..a586baea 100644 --- a/qadence/ml_tools/train_grad.py +++ b/qadence/ml_tools/train_grad.py @@ -66,10 +66,6 @@ def train( optimize_step: Customizable optimization callback which is called at every iteration.= The function must have the signature `optimize_step(model, optimizer, loss_fn, xs, device="cpu")`. - write_tensorboard: Customizable tensorboard logging callback which is - called every `config.write_every` iterations. The function must have - the signature `write_tensorboard(writer, loss, metrics, iteration)` - (see the example below). dtype: The dtype to use for the data. Example: