Ahmet Kaan Sever commited on
Commit
7a6ddbf
·
1 Parent(s): 2ac340e

fixed import

Browse files
Files changed (1) hide show
  1. src/deepeval/base_task.py +1 -1
src/deepeval/base_task.py CHANGED
@@ -2,7 +2,7 @@ from abc import ABC, abstractmethod
2
  from datasets import load_dataset
3
  import os
4
  from dotenv import load_dotenv
5
- from transformers import AutoModelForCausalLM, AutoTokenizer, LogitsProcessorList
6
  import torch
7
  from typing import List
8
  load_dotenv()
 
2
  from datasets import load_dataset
3
  import os
4
  from dotenv import load_dotenv
5
+ from transformers import AutoModelForCausalLM, AutoTokenizer, LogitsProcessorList, LogitsProcessor
6
  import torch
7
  from typing import List
8
  load_dotenv()