fix: add dependency

ServerlessLLM · Feb 21, 2025 · 5adb622 · 5adb622
1 parent 1c5c424
commit 5adb622
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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