From c1c14e1ce9a6ffe66079401d729dbf3b845d8983 Mon Sep 17 00:00:00 2001
From: Lillian-Violet <git@lillianviolet.dev>
Date: Wed, 4 Sep 2024 10:31:49 +0200
Subject: [PATCH] Also add azureml

---
 flake.nix                    |  1 +
 pkgs/azure-ai-ml/default.nix | 33 +++++++++++++++++++++++++++++++++
 2 files changed, 34 insertions(+)
 create mode 100644 pkgs/azure-ai-ml/default.nix

diff --git a/flake.nix b/flake.nix
index f6b0935..90c0262 100644
--- a/flake.nix
+++ b/flake.nix
@@ -39,6 +39,7 @@
             pydot
             graphviz
             (python311.pkgs.callPackage ./pkgs/azureml-mlflow/default.nix {})
+            (python311.pkgs.callPackage ./pkgs/azure-ai-ml/default.nix {})
           ]);
       };
     });
diff --git a/pkgs/azure-ai-ml/default.nix b/pkgs/azure-ai-ml/default.nix
new file mode 100644
index 0000000..e3de78e
--- /dev/null
+++ b/pkgs/azure-ai-ml/default.nix
@@ -0,0 +1,33 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchPypi,
+  setuptools,
+  python311,
+}:
+buildPythonPackage rec {
+  pname = "azure_ai_ml";
+  version = "1.19.0";
+  format = "wheel";
+
+  src = fetchPypi {
+    inherit pname version format;
+    sha256 = "sha256-8Dha8G776uH4MRNhPkU0NQjRKI/S8FhXYZ58fU1PUwI=";
+    dist = "py3";
+    python = "py3";
+  };
+
+  nativeBuildInputs = [setuptools];
+
+  propagatedBuildInputs = [
+  ];
+
+  doCheck = false; # Package does not contain tests
+
+  meta = with lib; {
+    description = "The azureml-mlflow package contains the integration code of AzureML with MLflow. MLflow (https://mlflow.org/) is an open-source platform for tracking machine learning experiments and managing models. You can use MLflow logging APIs with Azure Machine Learning so that metrics and artifacts are logged to your Azure machine learning workspace.";
+    homepage = "https://docs.microsoft.com/python/api/overview/azure/ml/?view=azure-ml-py";
+    license = licenses.mit;
+    maintainers = with maintainers; [Lillian-Violet];
+  };
+}