diff --git a/environment_setup/requirements.txt b/environment_setup/requirements.txt
index 8a086c4d..2d05c682 100644
--- a/environment_setup/requirements.txt
+++ b/environment_setup/requirements.txt
@@ -5,4 +5,5 @@ azureml-sdk>=1.0
 python-dotenv>=0.10.3
 flake8
 flake8_formatter_junit_xml
-azure-cli==2.0.71
\ No newline at end of file
+azure-cli==2.0.71
+azure-ml-api-sdk