diff --git a/pkgs/development/python-modules/mlflow-skinny/default.nix b/pkgs/development/python-modules/mlflow-skinny/default.nix index 7683c7c045fb..9f5a818c7a82 100644 --- a/pkgs/development/python-modules/mlflow-skinny/default.nix +++ b/pkgs/development/python-modules/mlflow-skinny/default.nix @@ -1,5 +1,7 @@ { + lib, buildPythonPackage, + fetchFromGitHub, mlflow, # build-system @@ -30,10 +32,17 @@ buildPythonPackage (finalAttrs: { pname = "mlflow-skinny"; - inherit (mlflow) version src; + inherit (mlflow) version; pyproject = true; __structuredAttrs = true; + src = fetchFromGitHub { + owner = "mlflow"; + repo = "mlflow"; + rev = "v${finalAttrs.version}"; + hash = "sha256-OxhM+KCem0sb9cwtyzrUD/MGfoiiCfgU47qipYRDaFk="; + }; + sourceRoot = "${finalAttrs.src.name}/libs/skinny"; build-system = [ setuptools ]; @@ -69,5 +78,6 @@ buildPythonPackage (finalAttrs: { meta = mlflow.meta // { description = "Lightweight version of MLflow that is designed to minimize package size"; homepage = "https://github.com/mlflow/mlflow/tree/master/libs/skinny"; + sourceProvenance = with lib.sourceTypes; [ fromSource ]; }; }) diff --git a/pkgs/development/python-modules/mlflow-tracing/default.nix b/pkgs/development/python-modules/mlflow-tracing/default.nix index ca76c7112bad..9f78d2dcaa32 100644 --- a/pkgs/development/python-modules/mlflow-tracing/default.nix +++ b/pkgs/development/python-modules/mlflow-tracing/default.nix @@ -1,6 +1,7 @@ { lib, buildPythonPackage, + fetchFromGitHub, mlflow, # build-system @@ -15,9 +16,6 @@ packaging, protobuf, pydantic, - - # tests - pytestCheckHook, }: buildPythonPackage (finalAttrs: { pname = "mlflow-tracing"; @@ -25,13 +23,16 @@ buildPythonPackage (finalAttrs: { pyproject = true; __structuredAttrs = true; - inherit (mlflow) src; + src = fetchFromGitHub { + owner = "mlflow"; + repo = "mlflow"; + rev = "v${finalAttrs.version}"; + hash = "sha256-OxhM+KCem0sb9cwtyzrUD/MGfoiiCfgU47qipYRDaFk="; + }; sourceRoot = "${finalAttrs.src.name}/libs/tracing"; - build-system = [ - setuptools - ]; + build-system = [ setuptools ]; dependencies = [ cachetools @@ -53,6 +54,9 @@ buildPythonPackage (finalAttrs: { description = "Open-Source SDK for observability and monitoring GenAI applications"; homepage = "https://github.com/mlflow/mlflow/tree/master/libs/tracing"; inherit (mlflow.meta) license; - maintainers = with lib.maintainers; [ GaetanLepage ]; + maintainers = with lib.maintainers; [ + GaetanLepage + gquetel + ]; }; })