diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml index e5cd5c6993..8f60a7b7f7 100644 --- a/.github/workflows/pytest.yml +++ b/.github/workflows/pytest.yml @@ -51,7 +51,9 @@ jobs: python -m pip install --progress-bar=off -r test_requirements.txt -c ${{ matrix.pip_constraints }} python -m pip install --progress-bar=off -e ./gym-unity -c ${{ matrix.pip_constraints }} - name: Save python dependencies - run: pip freeze > pip_versions-${{ matrix.python-version }}.txt + run: | + pip freeze > pip_versions-${{ matrix.python-version }}.txt + cat pip_versions-${{ matrix.python-version }}.txt - name: Run pytest run: pytest --cov=ml-agents --cov=ml-agents-envs --cov=gym-unity --cov-report html --junitxml=junit/test-results-${{ matrix.python-version }}.xml -p no:warnings - name: Upload pytest test results diff --git a/com.unity.ml-agents/CHANGELOG.md b/com.unity.ml-agents/CHANGELOG.md index 3bfe23ec52..62c0998c26 100755 --- a/com.unity.ml-agents/CHANGELOG.md +++ b/com.unity.ml-agents/CHANGELOG.md @@ -22,6 +22,7 @@ and this project adheres to #### com.unity.ml-agents (C#) - The Barracuda dependency was upgraded to 1.1.2 (#4571) #### ml-agents / ml-agents-envs / gym-unity (Python) +- The `action_probs` node is no longer listed as an output in TensorFlow models (#4613). ### Bug Fixes #### com.unity.ml-agents (C#) diff --git a/ml-agents/mlagents/trainers/tf/model_serialization.py b/ml-agents/mlagents/trainers/tf/model_serialization.py index c11f579215..aba884ed2c 100644 --- a/ml-agents/mlagents/trainers/tf/model_serialization.py +++ b/ml-agents/mlagents/trainers/tf/model_serialization.py @@ -40,9 +40,7 @@ ] ) -POSSIBLE_OUTPUT_NODES = frozenset( - ["action", "action_probs", "recurrent_out", "value_estimate"] -) +POSSIBLE_OUTPUT_NODES = frozenset(["action", "recurrent_out", "value_estimate"]) MODEL_CONSTANTS = frozenset( [ diff --git a/ml-agents/setup.py b/ml-agents/setup.py index fb834e8116..e9338ca45e 100644 --- a/ml-agents/setup.py +++ b/ml-agents/setup.py @@ -67,7 +67,8 @@ def run(self): # https://github.com/Unity-Technologies/ml-agents/blob/master/docs/Installation.md#windows-installing-pytorch 'torch>=1.6.0,<1.8.0;platform_system!="Windows"', "tensorboard>=1.15", - "cattrs>=1.0.0", + # cattrs 1.1.0 dropped support for python 3.6. + "cattrs>=1.0.0,<1.1.0", "attrs>=19.3.0", 'pypiwin32==223;platform_system=="Windows"', ],