improve error handling for missing hugging-face libraries

Author: jeremymanning

CITATION.cff

@@ -5,7 +5,7 @@ authors:
   given-names: "Jeremy"
   orcid: "https://orcid.org/0000-0001-7613-4732"
 title: "DataWrangler: wrangler your messy data into consistent well-organized formats"
-version: 0.2.1
+version: 0.2.2
 doi: 10.5281/zenodo.5123310
 date-released: 2022-07-25
 url: "https://github.com/ContextLab/data-wrangler"

HISTORY.rst

@@ -2,6 +2,16 @@
 History
 =======
 
+0.2.2 (22-07-25)
+-----------------
+
+* Better error handling when hugging-face libraries aren't installed and user asks to embed text using hugging-face models
+
+0.2.1 (22-07-25)
+------------------
+
+* Bug fixes when hugging-face libraries aren't installed
+
 0.2.0 (2022-07-25)
 ------------------
 

datawrangler/zoo/text.py

@@ -124,6 +124,9 @@ def model_lookup(model_name, parent):
             return eval(f'{parent}.{model_name}')
         except AttributeError:
             return None
+        except NameError:
+            raise ModuleNotFoundError('Hugging-face libraries have not been installed.  To use hugging-face models, please run "pip install --upgrade pydata-wrangler[hf]" to fix.')
+
 
     for p in ['text', 'decomposition', 'embeddings']:
         m = model_lookup(x, p)

setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.1
+current_version = 0.2.2
 commit = True
 tag = True
 

setup.py

@@ -46,6 +46,6 @@
     test_suite='tests',
     tests_require=test_requirements,
     url='https://github.com/ContextLab/data-wrangler',
-    version='0.2.1',
+    version='0.2.2',
     zip_safe=False,
 )