From e9babdcc37a9663bc4d94e0c48e6266de66694eb Mon Sep 17 00:00:00 2001 From: David Kyle Date: Wed, 25 Jan 2023 11:51:47 +0000 Subject: [PATCH] linting --- eland/ml/pytorch/nlp_ml_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/eland/ml/pytorch/nlp_ml_model.py b/eland/ml/pytorch/nlp_ml_model.py index 3de9e9b..6d84d43 100644 --- a/eland/ml/pytorch/nlp_ml_model.py +++ b/eland/ml/pytorch/nlp_ml_model.py @@ -123,6 +123,7 @@ class InferenceConfig: } } + class SlimInferenceOptions(InferenceConfig): def __init__( self, @@ -134,6 +135,7 @@ class SlimInferenceOptions(InferenceConfig): self.tokenization = tokenization self.results_field = results_field + class TextClassificationInferenceOptions(InferenceConfig): def __init__( self,