From af326d743a42a6047bba077dc1827aedddd45286 Mon Sep 17 00:00:00 2001
From: Mark Sandler <sandler@google.com>
Date: Fri, 19 Jul 2024 14:31:56 -0700
Subject: [PATCH] internal change

PiperOrigin-RevId: 654126390
---
 pytype/overlays/CMakeLists.txt    | 10 ++++++++++
 pytype/overlays/fiddle_overlay.py |  6 ++++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/pytype/overlays/CMakeLists.txt b/pytype/overlays/CMakeLists.txt
index b1bd1118b..5caccb897 100644
--- a/pytype/overlays/CMakeLists.txt
+++ b/pytype/overlays/CMakeLists.txt
@@ -336,3 +336,13 @@ py_library(
     pytype.pytd.pytd
     pytype.typegraph.cfg
 )
+
+py_library(
+  NAME
+    trail_configurable_overlay
+  SRCS
+    trail_configurable_overlay.py
+  DEPS
+    .fiddle_overlay
+    .overlay
+)
diff --git a/pytype/overlays/fiddle_overlay.py b/pytype/overlays/fiddle_overlay.py
index 1d944ba80..f774a224b 100644
--- a/pytype/overlays/fiddle_overlay.py
+++ b/pytype/overlays/fiddle_overlay.py
@@ -27,6 +27,7 @@
 _CLASS_ALIASES = {
     "Config": "Config",
     "PaxConfig": "Config",
+    "TemplateConfig": "Config",
     "Partial": "Partial",
     "PaxPartial": "Partial",
 }
@@ -331,11 +332,12 @@ def is_fiddle_buildable_pytd(cls: pytd.Class) -> bool:
   # fiddle._src.config.Config
   fiddle = re.fullmatch(r"fiddle\.(.+\.)?(Config|Partial)", cls.name)
   pax = re.fullmatch(r"(.+\.)?pax_fiddle.(Pax)?(Config|Partial)", cls.name)
-  return bool(fiddle or pax)
+  cfbl = re.fullmatch(r"(.+\.)?configurable.(Template)?(Config|)", cls.name)
+  return bool(fiddle or pax or cfbl)
 
 
 def get_fiddle_buildable_subclass(cls: pytd.Class) -> str:
-  if re.search(r"\.(Pax)?Config$", cls.name):
+  if re.search(r"\.(Pax|Template)?Config$", cls.name):
     return "Config"
   if re.search(r"\.(Pax)?Partial$", cls.name):
     return "Partial"