diff --git a/.yamato/com.unity.ml-agents-test.yml b/.yamato/com.unity.ml-agents-test.yml
index dc0aeca8287..6a62b4b85c6 100644
--- a/.yamato/com.unity.ml-agents-test.yml
+++ b/.yamato/com.unity.ml-agents-test.yml
@@ -8,7 +8,7 @@ test_editors:
       enableNoDefaultPackages: !!bool true
 
 trunk_editor:
-    - version: 6000.4
+    - version: trunk
       testProject: DevProject
 
 test_platforms:
diff --git a/.yamato/coverage_tests.metafile b/.yamato/coverage_tests.metafile
index e2d0ab0fbee..f329f6de674 100644
--- a/.yamato/coverage_tests.metafile
+++ b/.yamato/coverage_tests.metafile
@@ -6,7 +6,7 @@ coverage_test_platforms:
     - name: linux
       type: Unity::VM
       image: ml-agents/ubuntu-ci:v1.0.1
-      flavor: b1.medium
+      flavor: i1.xlarge
 
 coverage_test_packages:
     - name: com.unity.ml-agents
diff --git a/.yamato/test_versions.metafile b/.yamato/test_versions.metafile
index 2d4a1ee5d34..abcec74931c 100644
--- a/.yamato/test_versions.metafile
+++ b/.yamato/test_versions.metafile
@@ -7,5 +7,5 @@ test_editors:
     extra_test: gym
   - version: 6000.0
     extra_test: sensor
-  - version: 6000.3
+  - version: trunk
     extra_test: llapi
diff --git a/DevProject/Packages/manifest.json b/DevProject/Packages/manifest.json
index eed5e2ce770..f07b746c3ee 100644
--- a/DevProject/Packages/manifest.json
+++ b/DevProject/Packages/manifest.json
@@ -1,5 +1,6 @@
 {
   "dependencies": {
+    "com.unity.ai.inference": "https://github.cds.internal.unity3d.com/unity/UnityInferenceEngine.git?path=com.unity.ai.inference#dev",
     "com.unity.ai.navigation": "2.0.6",
     "com.unity.coding": "0.1.0-preview.13",
     "com.unity.inputsystem": "1.13.0",
diff --git a/Project/Packages/manifest.json b/Project/Packages/manifest.json
index 30298955dd4..1a8cc0d0564 100644
--- a/Project/Packages/manifest.json
+++ b/Project/Packages/manifest.json
@@ -1,5 +1,6 @@
 {
   "dependencies": {
+    "com.unity.ai.inference": "https://github.cds.internal.unity3d.com/unity/UnityInferenceEngine.git?path=com.unity.ai.inference#dev",
     "com.unity.ai.navigation": "2.0.6",
     "com.unity.formats.fbx": "5.1.1",
     "com.unity.ide.rider": "3.0.31",