diff --git a/compiler_gym/envs/gcc/gcc_env.py b/compiler_gym/envs/gcc/gcc_env.py index 3ca0b6246..245446659 100644 --- a/compiler_gym/envs/gcc/gcc_env.py +++ b/compiler_gym/envs/gcc/gcc_env.py @@ -16,8 +16,8 @@ from compiler_gym.envs.gcc.gcc_rewards import AsmSizeReward, ObjSizeReward from compiler_gym.service import ConnectionOpts from compiler_gym.service.client_service_compiler_env import ClientServiceCompilerEnv -from compiler_gym.spaces import Reward from compiler_gym.service.connection_pool import ServiceConnectionPoolBase +from compiler_gym.spaces import Reward from compiler_gym.util.decorators import memoized_property from compiler_gym.util.gym_type_hints import ObservationType, OptionalArgumentValue from compiler_gym.views import ObservationSpaceSpec diff --git a/tests/service/CMakeLists.txt b/tests/service/CMakeLists.txt index 6d1b77a5c..22d41faa9 100644 --- a/tests/service/CMakeLists.txt +++ b/tests/service/CMakeLists.txt @@ -15,6 +15,7 @@ if(COMPILER_GYM_ENABLE_LLVM_ENV) compiler_gym::compiler_gym compiler_gym::envs::envs compiler_gym::errors::errors + compiler_gym::service::service compiler_gym::service::service_cache tests::test_main ) diff --git a/tests/service/connection_pool_test.py b/tests/service/connection_pool_test.py index 62ee5d2f6..cb6773d39 100644 --- a/tests/service/connection_pool_test.py +++ b/tests/service/connection_pool_test.py @@ -8,8 +8,8 @@ import compiler_gym from compiler_gym.envs.llvm import LLVM_SERVICE_BINARY -from compiler_gym.service import ConnectionOpts, ServiceConnectionPool from compiler_gym.errors import ServiceError +from compiler_gym.service import ConnectionOpts, ServiceConnectionPool from tests.test_main import main pytest_plugins = ["tests.pytest_plugins.llvm"]