set up approval test file location

LittleLittleCloud · LittleLittleCloud · commit d6f0e617243a · 2024-07-19T10:57:55.000-07:00
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");
     }
 

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,10 @@ public class Phi2Tests`
`18`	`18`	`{`
`19`	`19`	`public Phi2Tests()`
`20`	`20`	`{`
	`21`	`+ if (Environment.GetEnvironmentVariable("HELIX_CORRELATION_ID") != null)`
	`22`	`+ {`
	`23`	`+ Approvals.UseAssemblyLocationForApprovedFiles();`
	`24`	`+ }`
`21`	`25`	`torch.set_default_device("meta");`
`22`	`26`	`}`
`23`	`27`
Original file line number	Diff line number	Diff line change
`@@ -19,6 +19,11 @@ public class Phi3Tests`
`19`	`19`	`{`
`20`	`20`	`public Phi3Tests()`
`21`	`21`	`{`
	`22`	`+ if (Environment.GetEnvironmentVariable("HELIX_CORRELATION_ID") != null)`
	`23`	`+ {`
	`24`	`+ Approvals.UseAssemblyLocationForApprovedFiles();`
	`25`	`+ }`
	`26`	`+`
`22`	`27`	`torch.set_default_device("meta");`
`23`	`28`	`}`
`24`	`29`