Move policy change to seal bench

Signed-off-by: Jakub Sztandera <[email protected]>
filecoin-project · Sep 25, 2020 · 4eec4a0 · 4eec4a0
1 parent 6db6a90
commit 4eec4a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cmd/lotus-bench/main.go b/cmd/lotus-bench/main.go
@@ -76,8 +76,6 @@ func main() {
 
 	log.Info("Starting lotus-bench")
 
-	policy.AddSupportedProofTypes(abi.RegisteredSealProof_StackedDrg2KiBV1)
-
 	app := &cli.App{
 		Name:    "lotus-bench",
 		Usage:   "Benchmark performance of lotus on your hardware",
@@ -147,6 +145,8 @@ var sealBenchCmd = &cli.Command{
 		},
 	},
 	Action: func(c *cli.Context) error {
+		policy.AddSupportedProofTypes(abi.RegisteredSealProof_StackedDrg2KiBV1)
+
 		if c.Bool("no-gpu") {
 			err := os.Setenv("BELLMAN_NO_GPU", "1")
 			if err != nil {