From 68a22a8001c558aa966c00b34f1c086ffa3046e2 Mon Sep 17 00:00:00 2001 From: David Kyle Date: Wed, 7 Jun 2023 12:34:53 +0100 Subject: [PATCH] Default the optional es_version parameter (#545) --- eland/ml/pytorch/transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eland/ml/pytorch/transformers.py b/eland/ml/pytorch/transformers.py index 89e4634..9b2b78e 100644 --- a/eland/ml/pytorch/transformers.py +++ b/eland/ml/pytorch/transformers.py @@ -578,7 +578,7 @@ class TransformerModel: model_id: str, task_type: str, *, - es_version: Optional[Tuple[int, int, int]], + es_version: Optional[Tuple[int, int, int]] = None, quantize: bool = False, ): """