set up approval test file location

dotnet · Jul 19, 2024 · d6f0e61 · d6f0e61
1 parent b316f19
commit d6f0e61
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs b/test/Microsoft.ML.GenAI.Phi.Tests/Phi2Tests.cs
@@ -18,6 +18,10 @@ public class Phi2Tests
 {
     public Phi2Tests()
     {
+        if (Environment.GetEnvironmentVariable("HELIX_CORRELATION_ID") != null)
+        {
+            Approvals.UseAssemblyLocationForApprovedFiles();
+        }
         torch.set_default_device("meta");
     }
 

diff --git a/test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs b/test/Microsoft.ML.GenAI.Phi.Tests/Phi3Tests.cs
@@ -19,6 +19,11 @@ public class Phi3Tests
 {
     public Phi3Tests()
     {
+        if (Environment.GetEnvironmentVariable("HELIX_CORRELATION_ID") != null)
+        {
+            Approvals.UseAssemblyLocationForApprovedFiles();
+        }
+
         torch.set_default_device("meta");
     }