diff --git a/constraints.txt b/constraints.txt
deleted file mode 100644
index 5eae777..0000000
--- a/constraints.txt
+++ /dev/null
@@ -1,18 +0,0 @@
-beartype==0.14.1
-collectibles==0.1.5
-einops==0.8.0
-jaxtyping==0.2.37
-matplotlib==3.10.0
-numpy==1.26.4
-openai==1.61.1
-pandas==2.2.3
-plotly==5.24.1
-pydantic==2.10.6
-scikit-learn==1.6.1
-seaborn==0.13.2
-tabulate==0.9.0
-torch==2.5.1
-torchvision==0.20.1
-tqdm==4.67.1
-transformer-lens==2.11.0
-sae-lens==5.4.1
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 8dd9635..af55324 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ classifiers = ["Topic :: Scientific/Engineering :: Artificial Intelligence"]
 
 [tool.poetry.dependencies]
 python = "^3.10"
-sae_lens = "^6.0.0"
+sae_lens = "^6.22.2"
 transformer-lens = ">=2.0.0"
 torch = ">=2.1.0"
 einops = ">=0.8.0"
@@ -32,7 +32,7 @@ matplotlib = ">=3.8.4"
 tabulate = ">=0.9.0"
 openai = ">=1.0.0"
 torchvision = ">=0.16.1" # required for what I believe are nnsight related issues
-datasets = "<4.0.0" # skylion openwebtext fails to load with datasets 4.0.0 currently, pending https://huggingface.co/datasets/Skylion007/openwebtext/discussions/22
+datasets = ">=3.0.0,<4.0.0" # skylion openwebtext fails to load with datasets 4.0.0 currently, pending https://huggingface.co/datasets/Skylion007/openwebtext/discussions/22
 
 # If running into dependency issues these are tested and working
 # [tool.poetry.dependencies]
@@ -63,7 +63,7 @@ ipykernel = ">=6.29.5"
 nbstripout = ">=0.7.1"
 loguru = ">=0.7.0"
 ruff = "^0.9.2"
-pyright = "^1.1.392.post0"
+pyright = "1.1.406"
 pre-commit = "^4.1.0"
 
 [tool.pyright]