From 16e47275677c95fae8711c9bcd13cfb81445e42c Mon Sep 17 00:00:00 2001 From: learned_optimization authors Date: Tue, 26 Mar 2024 04:51:15 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 619147712 --- docs/notebooks/summary_tutorial.ipynb | 2 +- docs/notebooks/summary_tutorial.md | 2 +- docs/notebooks/summary_tutorial.py | 3 ++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/docs/notebooks/summary_tutorial.ipynb b/docs/notebooks/summary_tutorial.ipynb index 41b3064..f84ee31 100644 --- a/docs/notebooks/summary_tutorial.ipynb +++ b/docs/notebooks/summary_tutorial.ipynb @@ -353,7 +353,7 @@ "source": [ "Then we can transform the `loss` function with the function transformation: `summary.with_summary_output_reduced`.\n", "This transformation goes through the computation and extracts all the tagged values and returns them to us by name in a dictionary.\n", - "In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/tensorflow/probability/blob/main/spinoffs/oryx/oryx/core/interpreters/harvest.py)).\n", + "In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/jax-ml/oryx/tree/main/oryx/core/interpreters/harvest.py)).\n", "When we wrap a function this, we return a tuple containing the original result, and a dictionary with the desired metrics." ] }, diff --git a/docs/notebooks/summary_tutorial.md b/docs/notebooks/summary_tutorial.md index 3c4c8a9..f0793ec 100644 --- a/docs/notebooks/summary_tutorial.md +++ b/docs/notebooks/summary_tutorial.md @@ -112,7 +112,7 @@ def loss(parameters): Then we can transform the `loss` function with the function transformation: `summary.with_summary_output_reduced`. This transformation goes through the computation and extracts all the tagged values and returns them to us by name in a dictionary. -In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/tensorflow/probability/blob/main/spinoffs/oryx/oryx/core/interpreters/harvest.py)). +In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/jax-ml/oryx/tree/main/oryx/core/interpreters/harvest.py)). When we wrap a function this, we return a tuple containing the original result, and a dictionary with the desired metrics. ```{code-cell} diff --git a/docs/notebooks/summary_tutorial.py b/docs/notebooks/summary_tutorial.py index 9a36604..6b008a5 100644 --- a/docs/notebooks/summary_tutorial.py +++ b/docs/notebooks/summary_tutorial.py @@ -25,6 +25,7 @@ # kernelspec: # display_name: Python 3 # name: python3 +# pylint: disable=line-too-long # --- # + [markdown] id="ryqPvTKI19zH" @@ -99,7 +100,7 @@ def loss(parameters): # + [markdown] id="AL9_xgfR4yPS" # Then we can transform the `loss` function with the function transformation: `summary.with_summary_output_reduced`. # This transformation goes through the computation and extracts all the tagged values and returns them to us by name in a dictionary. -# In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/tensorflow/probability/blob/main/spinoffs/oryx/oryx/core/interpreters/harvest.py)). +# In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/jax-ml/oryx/tree/main/oryx/core/interpreters/harvest.py)). # When we wrap a function this, we return a tuple containing the original result, and a dictionary with the desired metrics. # + colab={"base_uri": "https://localhost:8080/"} id="hZQkB6Um8PI5" outputId="984e4f64-7562-48ae-ca68-ff4014037553"