internal change

PiperOrigin-RevId: 654126390
google · Jul 22, 2024 · af326d7 · af326d7
1 parent 520f0a5
commit af326d7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git 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
@@ -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"