From 5ac8a053f0c64129e6a5a485083b98a806c82274 Mon Sep 17 00:00:00 2001 From: Youhei Sakurai Date: Fri, 7 Jul 2023 17:11:11 +0900 Subject: [PATCH] Fix No module named 'torch' (#553) Do not import torch unless necessary --- bin/eland_import_hub_model | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/bin/eland_import_hub_model b/bin/eland_import_hub_model index a4b6197..d0da973 100755 --- a/bin/eland_import_hub_model +++ b/bin/eland_import_hub_model @@ -31,7 +31,6 @@ import sys import tempfile import textwrap -import torch from elastic_transport.client_utils import DEFAULT from elasticsearch import AuthenticationException, Elasticsearch @@ -177,6 +176,8 @@ def check_cluster_version(es_client, logger): # PyTorch was upgraded to version 1.13.1 in 8.7. # and is incompatible with earlier versions if major_version == 8 and minor_version < 7: + import torch + logger.error(f"Eland uses PyTorch version {torch.__version__} which is incompatible with Elasticsearch versions prior to 8.7. Please upgrade Elasticsearch to at least version 8.7") exit(1)