Update example.ipynb

Signed-off-by: Zishen Wang <[email protected]>
materialsvirtuallab · Sep 18, 2024 · 6d3b16d · 6d3b16d
1 parent 8c5f549
commit 6d3b16d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/pes/qsnap/example.ipynb b/notebooks/pes/qsnap/example.ipynb
@@ -70,7 +70,7 @@
    "outputs": [],
    "source": [
     "from sklearn.linear_model import LinearRegression\n",
-    "\n",
+    "from maml.base import SKLModel\n",
     "from maml.apps.pes import SNAPotential\n",
     "from maml.describers import BispectrumCoefficients\n",
     "\n",