Update tests and examples with explicit SparsePoints location

lab-cosmo · Mar 23, 2021 · ac8002c · ac8002c
1 parent 7eede1a
commit ac8002c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/examples/MLIP_example.ipynb b/examples/MLIP_example.ipynb
@@ -43,7 +43,8 @@
     "import json\n",
     "\n",
     "from rascal.representations import SphericalInvariants\n",
-    "from rascal.models import Kernel, SparsePoints, train_gap_model\n",
+    "from rascal.models import Kernel, train_gap_model\n",
+    "from rascal.models.sparse_points import SparsePoints\n",
     "from rascal.models.IP_ase_interface import ASEMLCalculator\n",
     "from rascal.neighbourlist import AtomsList\n",
     "from rascal.utils import from_dict, to_dict, CURFilter, dump_obj, load_obj, get_score, print_score"

diff --git a/examples/Spherical_invariants_and_database_exploration.ipynb b/examples/Spherical_invariants_and_database_exploration.ipynb
@@ -45,7 +45,8 @@
     "from tqdm.notebook import tqdm\n",
     "\n",
     "from rascal.representations import SphericalInvariants\n",
-    "from rascal.models import Kernel, KRR, train_gap_model, SparsePoints\n",
+    "from rascal.models import Kernel, KRR, train_gap_model\n",
+    "from rascal.models.sparse_points import SparsePoints\n",
     "from rascal.utils import from_dict, to_dict, CURFilter, FPSFilter, get_score, print_score"
    ]
   },

diff --git a/tests/python/python_representation_calculator_test.py b/tests/python/python_representation_calculator_test.py
@@ -4,7 +4,8 @@
     SphericalInvariants,
 )
 from rascal.utils import from_dict, to_dict, FPSFilter
-from rascal.models import Kernel, SparsePoints
+from rascal.models import Kernel
+from rascal.models.sparse_points import SparsePoints
 from test_utils import load_json_frame, BoxList, Box, dot
 import unittest
 import numpy as np