From a4066224c26d76fe7a59c5c6f6c311d3ea5c935a Mon Sep 17 00:00:00 2001 From: Fahim Dalvi Date: Tue, 13 Aug 2024 00:28:38 +0700 Subject: [PATCH] Release v1.0.1 (#332) * Release v1.0.1 - Fix numpy version to be < 2 since some dependencies are incompatible with numpy >= 2 --- .github/workflows/publish-to-pypi.yml | 2 +- .github/workflows/publish-to-test-pypi.yml | 2 +- llmebench/datasets/ArMemes.py | 2 -- llmebench/datasets/dataset_base.py | 10 ++++++---- setup.cfg | 2 +- 5 files changed, 9 insertions(+), 9 deletions(-) diff --git a/.github/workflows/publish-to-pypi.yml b/.github/workflows/publish-to-pypi.yml index 28645050..c754d3fd 100644 --- a/.github/workflows/publish-to-pypi.yml +++ b/.github/workflows/publish-to-pypi.yml @@ -1,4 +1,4 @@ -name: Publish Python 🐍 distribution 📦 to PyPI and TestPyPI +name: Publish Python 🐍 distribution 📦 to PyPI on: push diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml index c2e97a17..c9e35328 100644 --- a/.github/workflows/publish-to-test-pypi.yml +++ b/.github/workflows/publish-to-test-pypi.yml @@ -1,4 +1,4 @@ -name: Publish Python 🐍 distribution 📦 to PyPI and TestPyPI +name: Publish Python 🐍 distribution 📦 to TestPyPI on: workflow_dispatch diff --git a/llmebench/datasets/ArMemes.py b/llmebench/datasets/ArMemes.py index 2df0cd4d..77be8fda 100644 --- a/llmebench/datasets/ArMemes.py +++ b/llmebench/datasets/ArMemes.py @@ -2,8 +2,6 @@ import json import os -from PIL import Image - from llmebench.datasets.dataset_base import DatasetBase from llmebench.tasks import TaskType diff --git a/llmebench/datasets/dataset_base.py b/llmebench/datasets/dataset_base.py index b5117cb3..9aab31bf 100644 --- a/llmebench/datasets/dataset_base.py +++ b/llmebench/datasets/dataset_base.py @@ -6,10 +6,6 @@ from abc import ABC, abstractmethod from pathlib import Path -from langchain.embeddings import HuggingFaceEmbeddings -from langchain.prompts.example_selector import MaxMarginalRelevanceExampleSelector -from langchain.vectorstores import FAISS - from pooch import Decompress, HTTPDownloader, Pooch, retrieve, Untar, Unzip import llmebench.utils as utils @@ -267,6 +263,12 @@ def prepare_fewshots( test sample """ + from langchain.embeddings import HuggingFaceEmbeddings + from langchain.prompts.example_selector import ( + MaxMarginalRelevanceExampleSelector, + ) + from langchain.vectorstores import FAISS + if embedding_model_name is None: embedding_model_name = "distiluse-base-multilingual-cased-v1" diff --git a/setup.cfg b/setup.cfg index 5ea93de3..ed18eb4a 100644 --- a/setup.cfg +++ b/setup.cfg @@ -1,6 +1,6 @@ [metadata] name = llmebench -version = 1.0.0 +version = 1.0.1 author = Fahim Dalvi author_email = faimaduddin@hbku.edu.qa description = A Flexible Framework for Accelerating LLMs Benchmarking