Add documentation for DistributedEmbedding.preprocess. (#2108)

hertschuh · web-flow · commit b437854a76c3 · 2025-05-21T18:33:09.000-07:00
Also remove spurious duplicate entry for `ModelParallel`.
diff --git a/scripts/api_master.py b/scripts/api_master.py
@@ -1905,13 +1905,6 @@
                         "keras.distribution.ModelParallel",
                     ],
                 },
-                {
-                    "path": "model_parallel",
-                    "title": "ModelParallel API",
-                    "generate": [
-                        "keras.distribution.ModelParallel",
-                    ],
-                },
                 {
                     "path": "distribution_utils",
                     "title": "Distribution utilities",
diff --git a/scripts/rs_master.py b/scripts/rs_master.py
@@ -92,6 +92,7 @@
             "generate": [
                 "keras_rs.layers.DistributedEmbedding",
                 "keras_rs.layers.DistributedEmbedding.call",
+                "keras_rs.layers.DistributedEmbedding.preprocess",
             ],
         },
         {

Original file line number	Diff line number	Diff line change
`@@ -92,6 +92,7 @@`
`92`	`92`	`"generate": [`
`93`	`93`	`"keras_rs.layers.DistributedEmbedding",`
`94`	`94`	`"keras_rs.layers.DistributedEmbedding.call",`
	`95`	`+ "keras_rs.layers.DistributedEmbedding.preprocess",`
`95`	`96`	`],`
`96`	`97`	`},`
`97`	`98`	`{`