Fix keras view source URLs

PiperOrigin-RevId: 630483598
tensorflow · May 3, 2024 · ca59b90 · ca59b90
1 parent 00ff355
commit ca59b90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/tools/docs/base_dir.py b/tensorflow/tools/docs/base_dir.py
@@ -32,10 +32,10 @@ def get_base_dirs_and_prefixes(code_url_prefix):
   base_dir = pathlib.Path(tf.__file__).parent
 
   if "dev" in tf.__version__:
-    keras_url_prefix = "https://github.com/keras-team/keras/tree/master/keras"
+    keras_url_prefix = "https://github.com/keras-team/keras/tree/master/keras/src"
   else:
     keras_url_prefix = (
-        f"https://github.com/keras-team/keras/tree/v{keras.__version__}/keras"
+        f"https://github.com/keras-team/keras/tree/v{keras.__version__}/keras/src"
     )
 
   if version.parse(tf.__version__) >= version.parse("2.16"):