BirdiD / BirdiDQ

BirdiDQ leverages the power of the Python Great Expectations open-source library and combines it with the simplicity of natural language queries to effortlessly identify and report data quality issues, all at the tip of your fingers.
Apache License 2.0
15 stars 2 forks source link

If CUDA support is not available, encoding call will fail silently #15

Closed elsatch closed 1 year ago

elsatch commented 1 year ago

Right now, when getting the expectations from natural language using the Falcon model, enconding call makes tokenizer bind to cuda:0 device. If no CUDA support is avaiable, this call will silently fail, providing no output.

Source code:

https://github.com/elsatch/BirdiDQ/blob/66940942e7405f373264c4f98a9ae3cafa7d321c/great_expectations/models/falcon_model.py#L41