From 03d9e8e6aa9a6f4176b39b2168ca0cc05f0e1410 Mon Sep 17 00:00:00 2001
From: Andrew Tolopko <atolopko@chanzuckerberg.com>
Date: Fri, 29 Jul 2022 09:57:55 -0400
Subject: [PATCH] fix unit test

avoid performing pip installs for mlflow-based tests
---
 test/unit/cli/test_annotate.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/unit/cli/test_annotate.py b/test/unit/cli/test_annotate.py
index bc0c1ca4..17f5127b 100644
--- a/test/unit/cli/test_annotate.py
+++ b/test/unit/cli/test_annotate.py
@@ -41,6 +41,9 @@ class TestCliAnnotate(unittest.TestCase):
                 model_file_path,
                 "--output-h5ad-file",
                 f"{query_dataset_file_path}.output",
+                # avoid having mflow create conda env or virtualenv when in test env;
+                # this avoids making pip remote requests and is also faster
+                "--mlflow-env-manager", "local"
             ],
         )