From c257d88b36478abc0ac105f586edc1ef97636b87 Mon Sep 17 00:00:00 2001
From: Mike <45373284+munkhuushmgl@users.noreply.github.com>
Date: Tue, 20 Apr 2021 15:58:12 -0700
Subject: [PATCH] test(env var): added missing requireEnv val (#255)

---
 ...teTrainingPipelineCustomTrainingManagedDatasetSampleTest.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aiplatform/snippets/src/test/java/aiplatform/CreateTrainingPipelineCustomTrainingManagedDatasetSampleTest.java b/aiplatform/snippets/src/test/java/aiplatform/CreateTrainingPipelineCustomTrainingManagedDatasetSampleTest.java
index ff1f16237cf..11cb9b8f1cd 100644
--- a/aiplatform/snippets/src/test/java/aiplatform/CreateTrainingPipelineCustomTrainingManagedDatasetSampleTest.java
+++ b/aiplatform/snippets/src/test/java/aiplatform/CreateTrainingPipelineCustomTrainingManagedDatasetSampleTest.java
@@ -56,6 +56,7 @@ private static void requireEnvVar(String varName) {
   @BeforeClass
   public static void checkRequirements() {
     requireEnvVar("GOOGLE_APPLICATION_CREDENTIALS");
+    requireEnvVar("CUSTOM_MANAGED_DATASET");
     requireEnvVar("UCAIP_PROJECT_ID");
   }