From cbc808bc6be539143e09d5ca485a64e3318f2532 Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Fri, 21 Feb 2020 04:02:17 +0100 Subject: [PATCH] [Tests] determine_tests_to_run.sh has a bug affecting RLlib testing to be skipped sometimes. (#7243) --- ci/travis/determine_tests_to_run.py | 5 +++-- rllib/agents/a3c/__init__.py | 6 +++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/ci/travis/determine_tests_to_run.py b/ci/travis/determine_tests_to_run.py index 4fbf36daa..bbdd1f13b 100644 --- a/ci/travis/determine_tests_to_run.py +++ b/ci/travis/determine_tests_to_run.py @@ -4,6 +4,7 @@ from __future__ import division from __future__ import print_function import os +import re import subprocess import sys from functools import partial @@ -54,13 +55,13 @@ if __name__ == "__main__": ] for changed_file in files: - if changed_file.startswith("python/ray/tune/"): + if changed_file.startswith("python/ray/tune"): RAY_CI_TUNE_AFFECTED = 1 RAY_CI_RLLIB_AFFECTED = 1 RAY_CI_RLLIB_FULL_AFFECTED = 1 RAY_CI_LINUX_WHEELS_AFFECTED = 1 RAY_CI_MACOS_WHEELS_AFFECTED = 1 - elif changed_file.startswith("python/ray/rllib/"): + elif re.match("^(python/ray/)?rllib/", changed_file): RAY_CI_RLLIB_AFFECTED = 1 RAY_CI_RLLIB_FULL_AFFECTED = 1 RAY_CI_LINUX_WHEELS_AFFECTED = 1 diff --git a/rllib/agents/a3c/__init__.py b/rllib/agents/a3c/__init__.py index 3d63eed0b..2be1ed9cc 100644 --- a/rllib/agents/a3c/__init__.py +++ b/rllib/agents/a3c/__init__.py @@ -7,6 +7,10 @@ A2CAgent = renamed_agent(A2CTrainer) A3CAgent = renamed_agent(A3CTrainer) __all__ = [ - "A2CAgent", "A3CAgent", "A2CTrainer", "A3CTrainer", "DEFAULT_CONFIG", + "A2CAgent", + "A3CAgent", + "A2CTrainer", + "A3CTrainer", + "DEFAULT_CONFIG", "A2CPipeline" ]