search

kathrinse / be_great

A novel approach for synthesizing tabular data using pretrained large language models
MIT License
276 stars 46 forks source link

No module named '_lzma' error on MacOS #53

Open romajain1 opened 3 months ago

romajain1 commented 3 months ago

`--------------------------------------------------------------------------- ModuleNotFoundError Traceback (most recent call last) Cell In[3], line 1 ----> 1 from be_great import GReaT 2 import pandas as pd 4 data = pd.read_csv('/Users/romajain/Development/inspect-ml/worker_compensation/stp_modelling/input/full_train_data_stage_2.csv')

File ~/.pyenv/versions/3.10.0/envs/stp_modelling_anomalies/lib/python3.10/site-packages/be_great/init.py:1 ----> 1 from .great import GReaT

File ~/.pyenv/versions/3.10.0/envs/stp_modelling_anomalies/lib/python3.10/site-packages/be_great/great.py:15 12 import torch 13 from transformers import AutoTokenizer, AutoModelForCausalLM, TrainingArguments ---> 15 from be_great.great_dataset import GReaTDataset, GReaTDataCollator 16 from be_great.great_start import ( 17 GReaTStart, 18 CategoricalStart, (...) 21 _pad_tokens, 22 ) 23 from be_great.great_trainer import GReaTTrainer

File ~/.pyenv/versions/3.10.0/envs/stp_modelling_anomalies/lib/python3.10/site-packages/be_great/great_dataset.py:4 1 import random 2 import typing as tp ... ---> 27 from _lzma import * 28 from _lzma import _encode_filter_properties, _decode_filter_properties 29 import _compression

ModuleNotFoundError: No module named '_lzma'`

This is my error on Python 3.10 on MacOS