From 5adb622eff416224878907d1166d039d6021c462 Mon Sep 17 00:00:00 2001 From: Yao Fu Date: Fri, 21 Feb 2025 14:08:36 +0000 Subject: [PATCH] fix: add dependency --- pyproject.toml | 2 ++ requirements.txt | 4 +++- tracestorm/data_loader.py | 2 +- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index e6edc77..1fc7b00 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,9 +12,11 @@ authors = [ ] dependencies = [ + "datasets>=3.3.2", "openai>=1.58.0", "numpy>=1.26.4", "pandas>=2.2.3", + "requests>=2.31.0", "seaborn>=0.13.2", "matplotlib>=3.9", "click>=8.1.8" diff --git a/requirements.txt b/requirements.txt index 2c61bb6..8f874c0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,8 @@ -datasets +click>=8.1.8 +datasets>=3.3.2 matplotlib>=3.9 numpy>=1.26.4 +openai>=1.58.0 pandas>=2.2.3 requests>=2.31.0 seaborn>=0.13.2 diff --git a/tracestorm/data_loader.py b/tracestorm/data_loader.py index 3348995..29eb79e 100644 --- a/tracestorm/data_loader.py +++ b/tracestorm/data_loader.py @@ -5,8 +5,8 @@ from typing import List, Optional, Tuple import pandas as pd - from datasets import load_dataset + from tracestorm.constants import DEFAULT_DATASET_FOLDER from tracestorm.logger import init_logger