Fix import name conflict.
diff --git a/compiler_opt/rl/inlining/__init__.py b/compiler_opt/rl/inlining/__init__.py
index f7be2ca..48fc0c2 100644
--- a/compiler_opt/rl/inlining/__init__.py
+++ b/compiler_opt/rl/inlining/__init__.py
@@ -16,7 +16,7 @@
import gin
-from compiler_opt.rl import env
+from compiler_opt.rl import env as mlgo_env
from compiler_opt.rl import problem_configuration
from compiler_opt.rl.inlining import config
from compiler_opt.rl.inlining import inlining_runner
@@ -27,7 +27,7 @@
class InliningConfig(problem_configuration.ProblemConfiguration):
"""Expose the regalloc eviction components."""
- def get_env(self) -> env.MLGOEnvironmentBase:
+ def get_env(self) -> mlgo_env.MLGOEnvironmentBase:
return inlining_env.get_inlining_env()
def get_runner_type(self):