fix pruner provide

celestiaorg · May 14, 2024 · 1c5532d · 1c5532d
1 parent 5aac80d
commit 1c5532d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/nodebuilder/pruner/module.go b/nodebuilder/pruner/module.go
@@ -75,8 +75,9 @@ func ConstructModule(tp node.Type, cfg *Config) fx.Option {
 			}),
 		)
 	case node.Light:
-		fx.Provide(light.NewPruner)
 		return fx.Module("prune",
+			baseComponents,
+			fx.Provide(light.NewPruner),
 			fx.Supply(light.Window),
 		)
 	default:

diff --git a/pruner/light/pruner.go b/pruner/light/pruner.go
@@ -8,6 +8,7 @@ import (
 	"github.com/ipfs/go-datastore"
 
 	"github.com/celestiaorg/celestia-node/header"
+	"github.com/celestiaorg/celestia-node/pruner"
 	"github.com/celestiaorg/celestia-node/share"
 	"github.com/celestiaorg/celestia-node/share/ipld"
 )
@@ -17,7 +18,7 @@ type Pruner struct {
 	ds    datastore.Datastore
 }
 
-func NewPruner(bstore blockstore.Blockstore, ds datastore.Datastore) *Pruner {
+func NewPruner(bstore blockstore.Blockstore, ds datastore.Batching) pruner.Pruner {
 	return &Pruner{bserv: blockservice.New(bstore, nil), ds: ds}
 }