Release 22/inconsistent device tensor action in trainers [Don't Merge]

.github/workflows/publish_pypi.yaml

@@ -35,7 +35,7 @@ jobs:
         python setup.py bdist_wheel
     - name: Publish distribution 📦 to Test PyPI
       if: startsWith(github.ref, 'refs/tags') && contains(github.ref, 'test')
-      uses: pypa/gh-action-pypi-publish@master
+      uses: pypa/gh-action-pypi-publish@release/v1
       with:
         password: ${{ secrets.TEST_PYPI_PASSWORD }}
         repository_url: https://test.pypi.org/legacy/

.yamato/pytest-gpu.yml

@@ -2,7 +2,7 @@ pytest_gpu:
   name: Pytest GPU
   agent:
     type: Unity::VM::GPU
-    image: ml-agents/ml-agents-ubuntu-18.04:latest
+    image: ml-agents/ubuntu-ci:v1.0.0
     flavor: b1.large
   commands:
     - |

.yamato/standalone-build-test.yml

@@ -5,7 +5,7 @@ test_linux_standalone_{{ editor.version }}_{{ editor.extra_test }}:
   name: Test Linux Standalone {{ editor.version }} {{ editor.extra_test }}
   agent:
     type: Unity::VM
-    image: ml-agents/ml-agents-ubuntu-18.04:latest
+    image: ml-agents/ubuntu-ci:v1.0.0
     flavor: b1.large
   variables:
     UNITY_VERSION: {{ editor.version }}

.yamato/wrench/api-validation-jobs.yml

@@ -0,0 +1,47 @@
+# Auto-generated by Recipe Engine, do not modify manually.
+# This job is generated by the wrench recipe engine module, see find the docs here: .
+
+# upm-ci validation tests for API Validation - ml-agents - 2023.2 - windows (2023.2 - Windows).
+api_validation_-_ml-agents_-_2023_2_-_windows:
+  name: API Validation - ml-agents - 2023.2 - windows
+  agent:
+    image: package-ci/win10:default
+    type: Unity::VM
+    flavor: b1.large
+  commands:
+  - command: curl %WRENCH_LOCAL_APV_DOWNLOAD_URL% -o wrench-localapv.zip
+  - command: 7z x wrench-localapv.zip
+  - command: python PythonScripts/print_machine_info.py
+  - command: npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
+    timeout: 20
+    retries: 10
+  - command: unity-downloader-cli -u 2023.2 -c Editor --fast
+    timeout: 10
+    retries: 3
+  - command: python PythonScripts/PackageJsonCondersor.py
+    timeout: 1
+    retries: 0
+  - command: upm-ci package test -u .Editor --package-path com.unity.ml-agents --type vetting-tests || exit 0
+    timeout: 30
+    retries: 0
+  - command: python PythonScripts/parse_upm_ci_results.py --package-path=com.unity.ml-agents
+    timeout: 2
+    retries: 0
+  artifacts:
+    Crash Dumps:
+      paths:
+      - CrashDumps/**
+    packages:
+      paths:
+      - upm-ci~/**/*
+  dependencies:
+  - path: .yamato/wrench/package-pack-jobs.yml#package_pack_-_ml-agents
+  variables:
+    UPMPVP_CONTEXT_WRENCH: 0.10.5.0
+  triggers:
+    expression: push.branch match "^release/.*"
+    cancel_old_ci: true
+  metadata:
+    Job Maintainers: '#rm-packageworks'
+    Wrench: 0.10.5.0
+

.yamato/wrench/package-pack-jobs.yml

@@ -0,0 +1,32 @@
+# Auto-generated by Recipe Engine, do not modify manually.
+# This job is generated by the wrench recipe engine module, see find the docs here: .
+
+# Pack and Sign ML Agents
+package_pack_-_ml-agents:
+  name: Package Pack - ml-agents
+  agent:
+    image: package-ci/ubuntu-20.04:default
+    type: Unity::VM
+    flavor: b1.large
+  commands:
+  - command: npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
+    timeout: 20
+    retries: 10
+  - command: curl $WRENCH_LOCAL_APV_DOWNLOAD_URL -o wrench-localapv.zip
+  - command: 7z x wrench-localapv.zip
+  - command: python PythonScripts/print_machine_info.py
+  - command: upm-ci package pack --package-path com.unity.ml-agents
+  - command: cp upm-ci~/packages/packages.json upm-ci~/packages/com.unity.ml-agents_packages.json
+  artifacts:
+    packages:
+      paths:
+      - upm-ci~/packages/**/*
+  variables:
+    UPMCI_ACK_LARGE_PACKAGE: 1
+    UPMPVP_CONTEXT_WRENCH: 0.10.5.0
+  metadata:
+    Job Maintainers: '#rm-packageworks'
+    Wrench: 0.10.5.0
+
+
+

.yamato/wrench/preview-a-p-v.yml

@@ -0,0 +1,164 @@
+# Auto-generated by Recipe Engine, do not modify manually.
+# This job is generated by the wrench recipe engine module, see find the docs here: .
+
+# Parent Preview APV Job.
+all_preview_apv_jobs:
+  name: All Preview APV Jobs
+  dependencies:
+  - path: .yamato/wrench/preview-a-p-v.yml#preview_apv_-_6000_0_-_macos
+  - path: .yamato/wrench/preview-a-p-v.yml#preview_apv_-_6000_0_-_ubuntu
+  - path: .yamato/wrench/preview-a-p-v.yml#preview_apv_-_6000_0_-_windows
+  triggers:
+    expression: push.branch match "^release/.*"
+    cancel_old_ci: true
+  metadata:
+    Job Maintainers: '#rm-packageworks'
+    Wrench: 0.10.5.0
+
+# Functional tests for dependents found in the latest 6000.0 manifest (MacOS).
+preview_apv_-_6000_0_-_macos:
+  name: Preview APV - 6000.0 - macos
+  agent:
+    image: package-ci/macos-13:default
+    type: Unity::VM::osx
+    flavor: b1.xlarge
+  commands:
+  - command: curl $WRENCH_LOCAL_APV_DOWNLOAD_URL -o wrench-localapv.zip
+  - command: 7z x wrench-localapv.zip
+  - command: python PythonScripts/print_machine_info.py
+  - command: npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
+    timeout: 20
+    retries: 10
+  - command: unity-downloader-cli -u 6000.0 -c Editor --fast
+    timeout: 10
+    retries: 3
+  - command: python PythonScripts/preview_apv.py --wrench-config=.yamato/wrench/wrench_config.json --editor-version=6000.0 --testsuite=editor,playmode --artifacts-path=PreviewApvArtifacts~
+  - command: echo 'Skipping Editor Manifest Validator as it is only supported on Windows'
+  artifacts:
+    Crash Dumps:
+      paths:
+      - CrashDumps/**
+    logs:
+      paths:
+      - '*.log'
+      - '*.xml'
+      - upm-ci~/test-results/**/*
+      - upm-ci~/temp/*/Logs/**
+      - upm-ci~/temp/*/Library/*.log
+      - upm-ci~/temp/*/*.log
+      - upm-ci~/temp/Builds/*.log
+    packages:
+      paths:
+      - upm-ci~/packages/**/*
+    PreviewAPVResults:
+      paths:
+      - PreviewApvArtifacts~/**
+    pvp-results:
+      paths:
+      - upm-ci~/pvp/**/*
+  dependencies:
+  - path: .yamato/wrench/package-pack-jobs.yml#package_pack_-_ml-agents
+  variables:
+    UPMPVP_CONTEXT_WRENCH: 0.10.5.0
+  metadata:
+    Job Maintainers: '#rm-packageworks'
+    Wrench: 0.10.5.0
+
+# Functional tests for dependents found in the latest 6000.0 manifest (Ubuntu).
+preview_apv_-_6000_0_-_ubuntu:
+  name: Preview APV - 6000.0 - ubuntu
+  agent:
+    image: package-ci/ubuntu-20.04:default
+    type: Unity::VM
+    flavor: b1.large
+  commands:
+  - command: curl $WRENCH_LOCAL_APV_DOWNLOAD_URL -o wrench-localapv.zip
+  - command: 7z x wrench-localapv.zip
+  - command: python PythonScripts/print_machine_info.py
+  - command: npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
+    timeout: 20
+    retries: 10
+  - command: unity-downloader-cli -u 6000.0 -c Editor --fast
+    timeout: 10
+    retries: 3
+  - command: python PythonScripts/preview_apv.py --wrench-config=.yamato/wrench/wrench_config.json --editor-version=6000.0 --testsuite=editor,playmode --artifacts-path=PreviewApvArtifacts~
+  - command: echo 'Skipping Editor Manifest Validator as it is only supported on Windows'
+  artifacts:
+    Crash Dumps:
+      paths:
+      - CrashDumps/**
+    logs:
+      paths:
+      - '*.log'
+      - '*.xml'
+      - upm-ci~/test-results/**/*
+      - upm-ci~/temp/*/Logs/**
+      - upm-ci~/temp/*/Library/*.log
+      - upm-ci~/temp/*/*.log
+      - upm-ci~/temp/Builds/*.log
+    packages:
+      paths:
+      - upm-ci~/packages/**/*
+    PreviewAPVResults:
+      paths:
+      - PreviewApvArtifacts~/**
+    pvp-results:
+      paths:
+      - upm-ci~/pvp/**/*
+  dependencies:
+  - path: .yamato/wrench/package-pack-jobs.yml#package_pack_-_ml-agents
+  variables:
+    UPMPVP_CONTEXT_WRENCH: 0.10.5.0
+  metadata:
+    Job Maintainers: '#rm-packageworks'
+    Wrench: 0.10.5.0
+
+# Functional tests for dependents found in the latest 6000.0 manifest (Windows).
+preview_apv_-_6000_0_-_windows:
+  name: Preview APV - 6000.0 - windows
+  agent:
+    image: package-ci/win10:default
+    type: Unity::VM
+    flavor: b1.large
+  commands:
+  - command: curl %WRENCH_LOCAL_APV_DOWNLOAD_URL% -o wrench-localapv.zip
+  - command: 7z x wrench-localapv.zip
+  - command: python PythonScripts/print_machine_info.py
+  - command: npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
+    timeout: 20
+    retries: 10
+  - command: unity-downloader-cli -u 6000.0 -c Editor --fast
+    timeout: 10
+    retries: 3
+  - command: python PythonScripts/preview_apv.py --wrench-config=.yamato/wrench/wrench_config.json --editor-version=6000.0 --testsuite=editor,playmode --artifacts-path=PreviewApvArtifacts~
+  - command: python PythonScripts/editor_manifest_validator.py --version=6000.0 --wrench-config=.yamato/wrench/wrench_config.json
+  artifacts:
+    Crash Dumps:
+      paths:
+      - CrashDumps/**
+    logs:
+      paths:
+      - '*.log'
+      - '*.xml'
+      - upm-ci~/test-results/**/*
+      - upm-ci~/temp/*/Logs/**
+      - upm-ci~/temp/*/Library/*.log
+      - upm-ci~/temp/*/*.log
+      - upm-ci~/temp/Builds/*.log
+    packages:
+      paths:
+      - upm-ci~/packages/**/*
+    PreviewAPVResults:
+      paths:
+      - PreviewApvArtifacts~/**
+    pvp-results:
+      paths:
+      - upm-ci~/pvp/**/*
+  dependencies:
+  - path: .yamato/wrench/package-pack-jobs.yml#package_pack_-_ml-agents
+  variables:
+    UPMPVP_CONTEXT_WRENCH: 0.10.5.0
+  metadata:
+    Job Maintainers: '#rm-packageworks'
+    Wrench: 0.10.5.0
+