Make validate_constructor_args local to pytorch handler

apache · Feb 10, 2023 · e5c9062 · e5c9062
1 parent fbdfe59
commit e5c9062
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/sdks/python/apache_beam/ml/inference/base.py b/sdks/python/apache_beam/ml/inference/base.py
@@ -174,12 +174,6 @@ def update_model_path(self, model_path: Optional[str] = None):
     """Update the model paths produced by side inputs."""
     pass
 
-  def validate_constructor_args(self):
-    """
-    Validate arguments passed to the ModelHandler constructor.
-    """
-    raise NotImplementedError
-
 
 class KeyedModelHandler(Generic[KeyT, ExampleT, PredictionT, ModelT],
                         ModelHandler[Tuple[KeyT, ExampleT],

diff --git a/sdks/python/apache_beam/ml/inference/pytorch_inference.py b/sdks/python/apache_beam/ml/inference/pytorch_inference.py
@@ -243,9 +243,6 @@ def __init__(
       self._batching_kwargs['max_batch_size'] = max_batch_size
     self._torch_script_model_path = torch_script_model_path
 
-    self.validate_constructor_args()
-
-  def validate_constructor_args(self):
     _validate_constructor_args(
         state_dict_path=self._state_dict_path,
         model_class=self._model_class,
@@ -462,10 +459,6 @@ def __init__(
     if max_batch_size is not None:
       self._batching_kwargs['max_batch_size'] = max_batch_size
     self._torch_script_model_path = torch_script_model_path
-
-    self.validate_constructor_args()
-
-  def validate_constructor_args(self):
     _validate_constructor_args(
         state_dict_path=self._state_dict_path,
         model_class=self._model_class,