Remove extra generate() def

jrmccluskey · jrmccluskey · commit dae57c562e19 · 2022-11-15T14:37:28.000-05:00
diff --git a/sdks/python/apache_beam/ml/inference/pytorch_inference_test.py b/sdks/python/apache_beam/ml/inference/pytorch_inference_test.py
@@ -182,10 +182,6 @@ def forward(self, x):
     out = self.linear(x)
     return {'output1': out, 'output2': out}
 
-  def generate(self, x):
-    out = self.linear(x)
-    return {'output1': out, 'output2': out}
-
 
 class PytorchLinearRegressionKeyedBatchAndExtraInferenceArgs(torch.nn.Module):
   """
