fix doc

riteshghorse · riteshghorse · commit 627fdd931d89 · 2023-02-15T10:33:51.000-05:00
diff --git a/sdks/python/apache_beam/ml/inference/tensorflow_inference.py b/sdks/python/apache_beam/ml/inference/tensorflow_inference.py
@@ -100,10 +100,9 @@ def __init__(
 
     Args:
         model_uri (str): path to the trained model.
-        model_type (ModelType): type of model to be loaded.
-          Defaults to SAVED_MODEL.
-        inference_fn (TensorInferenceFn, Optional): inference function to use
-          during RunInference. Defaults to default_numpy_inference_fn.
+        model_type: type of model to be loaded. Defaults to SAVED_MODEL.
+        inference_fn: inference function to use during RunInference.
+          Defaults to default_numpy_inference_fn.
 
     **Supported Versions:** RunInference APIs in Apache Beam have been tested
     with Tensorflow 2.9, 2.10, 2.11.
@@ -182,10 +181,10 @@ def __init__(
 
     Args:
         model_uri (str): path to the trained model.
-        model_type (ModelType): type of model to be loaded.
+        model_type: type of model to be loaded.
           Defaults to SAVED_MODEL.
-        inference_fn (TensorInferenceFn, Optional): inference function to use
-          during RunInference. Defaults to default_numpy_inference_fn.
+        inference_fn: inference function to use during RunInference.
+          Defaults to default_numpy_inference_fn.
 
     **Supported Versions:** RunInference APIs in Apache Beam have been tested
     with Tensorflow 2.11.
