Fix signature generation when the method is dataclass instance.

PiperOrigin-RevId: 619368090
tensorflow · Mar 27, 2024 · ff989f0 · ff989f0
1 parent b647684
commit ff989f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/tensorflow_docs/api_generator/signature.py b/tools/tensorflow_docs/api_generator/signature.py
@@ -580,7 +580,7 @@ def generate_signature(
 
   sig = sig.replace(parameters=params)
 
-  if dataclasses.is_dataclass(func):
+  if dataclasses.is_dataclass(func) and inspect.isclass(func):
     sig = sig.replace(return_annotation=EMPTY)
     extract_fn = _extract_class_defaults_and_annotations
   else: