ctheodoris

/

Geneformer

like 216

Hello, after pulling the latest version, I receive the following error when importing the Geneformer Classifier. I believe including peft in the requirements.txt and setup.py files may solve this issue.

Code:

from geneformer import Classifier

Error:

File ~/.conda/envs/geneformer_env/lib/python3.10/site-packages/geneformer/perturber_utils.py:12
10 import torch
11 from datasets import Dataset, load_from_disk
---> 12 from peft import LoraConfig, get_peft_model
13 from transformers import (
14 BertForMaskedLM,
15 BertForSequenceClassification,
16 BertForTokenClassification,
17 BitsAndBytesConfig,
18 )
20 GENE_MEDIAN_FILE = Path(file).parent / "gene_median_dictionary.pkl"

ModuleNotFoundError: No module named 'peft'