diff --git a/pkgs/development/python-modules/nianet/default.nix b/pkgs/development/python-modules/nianet/default.nix
index 5a9ca788238e..bcdcb08bb9df 100644
--- a/pkgs/development/python-modules/nianet/default.nix
+++ b/pkgs/development/python-modules/nianet/default.nix
@@ -27,18 +27,22 @@ buildPythonPackage rec {
     sha256 = "sha256-FZipl6Z9AfiL6WH0kvUn8bVxt8JLdDVlmTSqnyxe0nY=";
   };
 
-  nativeBuildInputs = [
+  build-system = [
     poetry-core
     toml-adapt
   ];
 
-  propagatedBuildInputs = [
+  dependencies = [
     niapy
     numpy
     scikit-learn
     torch
   ];
 
+  pythonRelaxDeps = [
+    "numpy"
+  ];
+
   # create niapy and torch dep version consistent
   preBuild = ''
     toml-adapt -path pyproject.toml -a change -dep niapy -ver X