From b300ce980abb0af3c121c410cc83a544d9ad0422 Mon Sep 17 00:00:00 2001 From: Fabian Pedregosa Date: Fri, 9 Feb 2024 14:26:22 +0000 Subject: [PATCH 1/4] Add missing examples to contrib examples section. Also add flax as dependency to examples, which should fix the error in https://optax.readthedocs.io/en/latest/_collections/examples/mlp_mnist.html --- examples/contrib/README.md | 2 +- pyproject.toml | 8 +++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/examples/contrib/README.md b/examples/contrib/README.md index 7b56fad2..dea70274 100644 --- a/examples/contrib/README.md +++ b/examples/contrib/README.md @@ -7,5 +7,5 @@ Examples that make use of the `optax.contrib` module. :glob: :maxdepth: 1 -[!README.md]* +* ``` diff --git a/pyproject.toml b/pyproject.toml index 3480f8fe..138ad2de 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,8 +38,8 @@ dependencies = [ ] [project.urls] -homepage = "https://github.com/deepmind/optax" -repository = "https://github.com/deepmind/optax" +homepage = "https://github.com/google-deepmind/optax" +repository = "https://github.com/google-deepmind/optax" documentation = "https://optax.readthedocs.io/" [project.optional-dependencies] @@ -51,7 +51,8 @@ test = [ examples = [ "tensorflow-datasets>=4.2.0", "tensorflow>=2.4.0", - "dp_accounting>=0.4" + "dp_accounting>=0.4", + "flax", ] docs = [ @@ -66,6 +67,7 @@ docs = [ "sphinx-collections>=0.0.1", "tensorflow>=2.4.0", "tensorflow-datasets>=4.2.0", + "flax", ] dp-accounting = [ From 792109478717bf623274605f6a67cb764e988a39 Mon Sep 17 00:00:00 2001 From: Fabian Pedregosa Date: Fri, 9 Feb 2024 15:12:19 +0000 Subject: [PATCH 2/4] remove dangling comment --- examples/README.md | 1 - examples/contrib/README.md | 1 - 2 files changed, 2 deletions(-) diff --git a/examples/README.md b/examples/README.md index 13946ce7..a5bd5baf 100644 --- a/examples/README.md +++ b/examples/README.md @@ -2,7 +2,6 @@ This directory contains examples using the optax library. - ```{toctree} :glob: :maxdepth: 1 diff --git a/examples/contrib/README.md b/examples/contrib/README.md index dea70274..54c380b8 100644 --- a/examples/contrib/README.md +++ b/examples/contrib/README.md @@ -2,7 +2,6 @@ Examples that make use of the `optax.contrib` module. - ```{toctree} :glob: :maxdepth: 1 From 34cd492525cf3c6dc8e41af5f2a25cf027b93b75 Mon Sep 17 00:00:00 2001 From: Fabian Pedregosa Date: Fri, 9 Feb 2024 15:17:12 +0000 Subject: [PATCH 3/4] Deepmind -> Google Deepmind --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 138ad2de..98e28e50 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -10,7 +10,7 @@ readme = "README.md" license = { file = "LICENSE" } requires-python = ">=3.9" authors = [ - {name = "DeepMind", email = "optax-dev@google.com"}, + {name = "Google DeepMind", email = "optax-dev@google.com"}, ] keywords = [ "python", From fa93ae014ab8303b2b4db7d7e12bf74a74a271d2 Mon Sep 17 00:00:00 2001 From: Fabian Pedregosa Date: Fri, 9 Feb 2024 15:23:14 +0000 Subject: [PATCH 4/4] remove dangling spaces --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 98e28e50..f4d72261 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,8 +13,8 @@ authors = [ {name = "Google DeepMind", email = "optax-dev@google.com"}, ] keywords = [ - "python", - "machine learning", + "python", + "machine learning", "reinforcement-learning" ] classifiers = [