summaryrefslogtreecommitdiff
path: root/main/src/main/MainRunner.scala
diff options
context:
space:
mode:
Diffstat (limited to 'main/src/main/MainRunner.scala')
-rw-r--r--main/src/main/MainRunner.scala5
1 files changed, 4 insertions, 1 deletions
diff --git a/main/src/main/MainRunner.scala b/main/src/main/MainRunner.scala
index 6705a4b3..354b6173 100644
--- a/main/src/main/MainRunner.scala
+++ b/main/src/main/MainRunner.scala
@@ -9,6 +9,7 @@ import mill.eval.{Evaluator, PathRef}
import mill.util.PrintLogger
import scala.annotation.tailrec
+import ammonite.runtime.ImportHook
/**
@@ -120,11 +121,13 @@ class MainRunner(val config: ammonite.main.Cli.Config,
}
override def initMain(isRepl: Boolean) = {
+ val hooks = ImportHook.defaults + (Seq("ivy") -> MillIvyHook)
super.initMain(isRepl).copy(
scriptCodeWrapper = CustomCodeWrapper,
// Ammonite does not properly forward the wd from CliConfig to Main, so
// force forward it outselves
- wd = config.wd
+ wd = config.wd,
+ importHooks = hooks
)
}