diff --git a/pkgs/development/python-modules/compressai/default.nix b/pkgs/development/python-modules/compressai/default.nix
index 79de2d7596fd..6e453e9488e7 100644
--- a/pkgs/development/python-modules/compressai/default.nix
+++ b/pkgs/development/python-modules/compressai/default.nix
@@ -10,14 +10,17 @@
 
   # dependencies
   einops,
-  numpy,
   matplotlib,
+  numpy,
   pandas,
   pytorch-msssim,
   scipy,
+  tomli,
   torch,
   torch-geometric,
   torchvision,
+  tqdm,
+  typing-extensions,
 
   # optional-dependencies
   ipywidgets,
@@ -30,14 +33,14 @@
 
 buildPythonPackage rec {
   pname = "compressai";
-  version = "1.2.6";
+  version = "1.2.8";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "InterDigitalInc";
     repo = "CompressAI";
     tag = "v${version}";
-    hash = "sha256-xvzhhLn0iBzq3h1nro8/83QWEQe9K4zRa3RSZk+hy3Y=";
+    hash = "sha256-Fgobh7Q1rKomcqAT4kJl2RsM1W13ErO8sFB2urCqrCk=";
     fetchSubmodules = true;
   };
 
@@ -46,16 +49,22 @@ buildPythonPackage rec {
     setuptools
   ];
 
+  pythonRelaxDeps = [
+    "numpy"
+  ];
   dependencies = [
     einops
-    numpy
     matplotlib
+    numpy
     pandas
     pytorch-msssim
     scipy
+    tomli
     torch
     torch-geometric
     torchvision
+    tqdm
+    typing-extensions
   ];
 
   optional-dependencies = {
@@ -70,11 +79,9 @@ buildPythonPackage rec {
     "compressai._CXX"
   ];
 
+  # We have to delete the source because otherwise it is used intead the installed package.
   preCheck = ''
-    # We have to delete the source because otherwise it is used intead the installed package.
     rm -rf compressai
-
-    export HOME=$(mktemp -d)
   '';
 
   nativeCheckInputs = [
@@ -91,6 +98,7 @@ buildPythonPackage rec {
     "test_pretrained"
 
     # Flaky (AssertionError: assert 0.08889999999999998 < 0.064445)
+    "test_compiling"
     "test_find_close"
   ];