third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

linting

Browse Source
This commit is contained in:
David Kyle 2023-01-25 11:51:47 +00:00
parent 3a044a76e3
commit e9babdcc37
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
eland/ml/pytorch/nlp_ml_model.py
View File

@ -123,6 +123,7 @@ class InferenceConfig:
} }
} }
class SlimInferenceOptions(InferenceConfig): class SlimInferenceOptions(InferenceConfig):
def __init__( def __init__(
self, self,
@ -134,6 +135,7 @@ class SlimInferenceOptions(InferenceConfig):
self.tokenization = tokenization self.tokenization = tokenization
self.results_field = results_field self.results_field = results_field
class TextClassificationInferenceOptions(InferenceConfig): class TextClassificationInferenceOptions(InferenceConfig):
def __init__( def __init__(
self, self,