Simplify determining the root module

It is always the first one.
tweag · Sep 1, 2023 · 8a5350e · 8a5350e
1 parent 818c288
commit 8a5350e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extensions/ghc_version.bzl b/extensions/ghc_version.bzl
@@ -7,9 +7,9 @@ load(
 )
 
 def _ghc_default_version_impl(mctx):
-    root_module = [ module.name for module in mctx.modules if module.is_root ]
+    root_module = mctx.modules[0].name
 
-    if root_module and root_module[0] in ['rules_haskell', 'rules_haskell_tests']:
+    if root_module in ['rules_haskell', 'rules_haskell_tests']:
         _ghc_version(name = "rules_haskell_ghc_version")
     else:
         _ghc_default_version(name = "rules_haskell_ghc_version")