fix pydoc

apache · Jul 20, 2023 · 1a022d9 · 1a022d9
1 parent 467e5dd
commit 1a022d9
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/sdks/python/apache_beam/ml/inference/huggingface_inference.py b/sdks/python/apache_beam/ml/inference/huggingface_inference.py
@@ -178,8 +178,8 @@ def __init__(
     Use HuggingFaceModelHandlerKeyedTensor or HuggingFaceModelHandlerTensor.
 
     Example Usage model::
-    pcoll | RunInference(HuggingFaceModelHandlerKeyedTensor(
-      model_uri="bert-base-uncased", model_class=AutoModelForMaskedLM))
+      pcoll | RunInference(HuggingFaceModelHandlerKeyedTensor(
+        model_uri="bert-base-uncased", model_class=AutoModelForMaskedLM))
 
     Args:
       model_uri (str): path to the pretrained model on the hugging face
@@ -266,8 +266,8 @@ class HuggingFaceModelHandlerKeyedTensor(
   automatically.
 
   Example Usage model::
-  pcoll | RunInference(HuggingFaceModelHandlerKeyedTensor(
-    model_uri="bert-base-uncased", model_class=AutoModelForMaskedLM))
+    pcoll | RunInference(HuggingFaceModelHandlerKeyedTensor(
+      model_uri="bert-base-uncased", model_class=AutoModelForMaskedLM))
 
   **Supported Versions:** HuggingFaceModelHandler supports transformers>=4.18.0.
   """
@@ -367,8 +367,8 @@ class HuggingFaceModelHandlerTensor(HuggingFaceModelHandler[Union[tf.Tensor,
   automatically.
 
   Example Usage model:
-  pcoll | RunInference(HuggingFaceModelHandlerTensor(
-    model_uri="bert-base-uncased", model_class=AutoModelForMaskedLM))
+    pcoll | RunInference(HuggingFaceModelHandlerTensor(
+      model_uri="bert-base-uncased", model_class=AutoModelForMaskedLM))
 
   **Supported Versions:** HuggingFaceModelHandler supports transformers>=4.18.0.
   """