From 34144f9e0c96c251c5ab50a88f6f10df0971c5e9 Mon Sep 17 00:00:00 2001 From: Anda Date: Thu, 9 Nov 2023 09:19:29 -0800 Subject: [PATCH] chore: fix profiler test in CI --- e2e_tests/tests/fixtures/mnist_pytorch/profiling.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/e2e_tests/tests/fixtures/mnist_pytorch/profiling.py b/e2e_tests/tests/fixtures/mnist_pytorch/profiling.py index 4d017a787e1..6d71cd0daea 100644 --- 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)