chore: fix profiler test in CI

determined-ai · Nov 9, 2023 · 34144f9 · 34144f9
1 parent 1d64941
commit 34144f9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/e2e_tests/tests/fixtures/mnist_pytorch/profiling.py b/e2e_tests/tests/fixtures/mnist_pytorch/profiling.py
@@ -4,6 +4,7 @@
 
 import determined as det
 from determined import pytorch
+from determined.common.api import certs
 
 
 def run():
@@ -12,6 +13,9 @@ def run():
     info = det.get_cluster_info()
     assert info, "Test must be run on cluster."
 
+    # TODO: refactor profiling to to not use the cli_cert.
+    certs.cli_cert = certs.default_load(info.master_url)
+
     with pytorch.init() as train_context:
         trial = mnist_pytorch.MNistTrial(train_context, hparams=info.trial.hparams)
         trainer = pytorch.Trainer(trial, train_context)