From b2f91e29c6e5ab9d1c05f7a703bef609286e3979 Mon Sep 17 00:00:00 2001 From: Steve Lee Date: Tue, 11 Jul 2023 11:59:12 -0700 Subject: [PATCH 1/2] Expand java extractor capability for InputSelectorDef and TensorEncoderDef --- .../buildgen/javadefref/JavaSourceEntityExtractor.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala b/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala index 2d9dc21b..b70093ba 100644 --- a/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala +++ b/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala @@ -61,7 +61,7 @@ object JavaSourceEntityExtractor { // TODO there needs to be some way to configure this val generators: Set[String] = - Set("InputTransformDef", "FeatureEncoderDef", "DataTransformDef") + Set("InputTransformDef", "FeatureEncoderDef", "DataTransformDef", "TensorEncoderDef", "InputSelectorDef") val topLevelDefsTypes: SortedSet[Entity] = compUnit From 43459da4802f146a13729140cff449c26664f292 Mon Sep 17 00:00:00 2001 From: Steve Lee Date: Tue, 11 Jul 2023 12:19:50 -0700 Subject: [PATCH 2/2] Format file --- .../buildgen/javadefref/JavaSourceEntityExtractor.scala | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala b/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala index b70093ba..f6420014 100644 --- a/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala +++ b/language_generators/scala-defref-extractor/src/main/scala/io/bazeltools/buildgen/javadefref/JavaSourceEntityExtractor.scala @@ -61,7 +61,13 @@ object JavaSourceEntityExtractor { // TODO there needs to be some way to configure this val generators: Set[String] = - Set("InputTransformDef", "FeatureEncoderDef", "DataTransformDef", "TensorEncoderDef", "InputSelectorDef") + Set( + "InputTransformDef", + "FeatureEncoderDef", + "DataTransformDef", + "TensorEncoderDef", + "InputSelectorDef" + ) val topLevelDefsTypes: SortedSet[Entity] = compUnit