diff --git a/integration/feature/inspect/src/InspectTests.scala b/integration/feature/inspect/src/InspectTests.scala index 7b63e877c48a..82ea1c0454ec 100644 --- a/integration/feature/inspect/src/InspectTests.scala +++ b/integration/feature/inspect/src/InspectTests.scala @@ -146,7 +146,6 @@ object InspectTests extends UtestIntegrationTestSuite { | mill.javalib.bsp.BspModule | mill.javalib.RunModule | mill.javalib.GenIdeaModule - | mill.javalib.OfflineSupportModule | mill.javalib.SemanticDbJavaModule | mill.javalib.AssemblyModule | mill.javalib.JavaModule @@ -171,7 +170,6 @@ object InspectTests extends UtestIntegrationTestSuite { | mill.javalib.bsp.BspModule | mill.javalib.RunModule | mill.javalib.GenIdeaModule - | mill.javalib.OfflineSupportModule | mill.javalib.SemanticDbJavaModule | mill.javalib.AssemblyModule | mill.javalib.JavaModule diff --git a/libs/javalib/src/mill/javalib/AssemblyModule.scala b/libs/javalib/src/mill/javalib/AssemblyModule.scala index a3ef4a185f4d..6573ab4b305f 100644 --- a/libs/javalib/src/mill/javalib/AssemblyModule.scala +++ b/libs/javalib/src/mill/javalib/AssemblyModule.scala @@ -11,7 +11,7 @@ import mill.util.Jvm /** * Module that provides functionality around creating and configuring JVM assembly jars */ -trait AssemblyModule extends mill.api.Module { +trait AssemblyModule extends mill.api.Module with OfflineSupportModule { outer => def finalMainClassOpt: T[Either[String, String]] @@ -148,6 +148,11 @@ trait AssemblyModule extends mill.api.Module { created.pathRef } } + + override def prepareOffline(all: mainargs.Flag): Task.Command[Seq[PathRef]] = Task.Command { + AssemblyModule.prepareOffline(all)() + } + } object AssemblyModule extends ExternalModule with CoursierModule with OfflineSupportModule {