materialyzeai
diff --git a/‎examples/Running Softening Benchmark.ipynb‎
Lines changed: 4 additions & 1 deletion b/‎examples/Running Softening Benchmark.ipynb‎
Lines changed: 4 additions & 1 deletion
@@ -30,7 +30,10 @@
     "import pandas as pd\n",
     "\n",
     "from matcalc import load_fp\n",
-    "from matcalc.benchmark import SofteningBenchmark"
+    "from matcalc.benchmark import SofteningBenchmark\n",
+    "\n",
+    "import matgl\n",
+    "matgl.set_backend(\"DGL\")"
    ]
   },
   {
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,10 @@`
`30`	`30`	`"import pandas as pd\n",`
`31`	`31`	`"\n",`
`32`	`32`	`"from matcalc import load_fp\n",`
`33`		`- "from matcalc.benchmark import SofteningBenchmark"`
	`33`	`+ "from matcalc.benchmark import SofteningBenchmark\n",`
	`34`	`+ "\n",`
	`35`	`+ "import matgl\n",`
	`36`	`+ "matgl.set_backend(\"DGL\")"`
`34`	`37`	`]`
`35`	`38`	`},`
`36`	`39`	`{`