From 76c3bb77dcaba3fd6b8f69ca02f9d06c0846b78b Mon Sep 17 00:00:00 2001 From: Sujeevan Rasaratnam Date: Wed, 22 May 2024 01:10:53 +0000 Subject: [PATCH] disable install from s3 because blis, en_core_web_sm, etc. are not in s3 yet --- transformers/nlp/text_named_entities_transformer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/transformers/nlp/text_named_entities_transformer.py b/transformers/nlp/text_named_entities_transformer.py index 8e23f34c..dd6c908f 100644 --- a/transformers/nlp/text_named_entities_transformer.py +++ b/transformers/nlp/text_named_entities_transformer.py @@ -12,10 +12,11 @@ class TextNamedEntityTransformer(CustomTransformer): """Transformer to extract the count of Named Entities""" _testing_can_skip_failure = False # ensure tested as if shouldn't fail _root_path = "https://s3.amazonaws.com/artifacts.h2o.ai/deps/dai/recipes" - _suffix = "-cp38-cp38-linux_x86_64.whl" - froms3 = True + _suffix = "-cp311-cp311-linux_x86_64.whl" + froms3 = False _is_reproducible = False # some issue with deepcopy and refit, do not get same result if froms3: + # TODO: upload the wheel files to S3 _modules_needed_by_name = [ '%s/blis-0.4.1%s' % (_root_path, _suffix), '%s/catalogue-1.0.0%s' % (_root_path, _suffix),