From 1ae879cc1b1b970fc691cdba9b80cd3fcf4fadee Mon Sep 17 00:00:00 2001 From: Max Veit Date: Tue, 23 Mar 2021 15:33:50 +0100 Subject: [PATCH] Update tests and examples with explicit SparsePoints location --- examples/MLIP_example.ipynb | 3 ++- examples/Spherical_invariants_and_database_exploration.ipynb | 3 ++- tests/python/python_representation_calculator_test.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/examples/MLIP_example.ipynb b/examples/MLIP_example.ipynb index d87e018cf..dcb0b31b4 100644 --- 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 index a72977f8b..dfd31dc17 100644 --- 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 index f8d6e4b12..e71de807e 100644 --- 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