update

Author: gengxingze

abfml/data/__init__.py

@@ -1,4 +1,4 @@
-from .read_data import ReadData
-from .dataset import ABFMLDataset
+from abfml.data.read_data import ReadData
+from abfml.data.dataset import ABFMLDataset
 
 __all__ = ["read_data", "ReadData", "ABFMLDataset"]

abfml/data/read_data.py

@@ -2,12 +2,11 @@
 import numpy as np
 from concurrent.futures import ProcessPoolExecutor
 from functools import partial
-from ase import Atoms
-from ase import io
+from ase import Atoms, io
 from ase.neighborlist import NeighborList
 from typing import List, Optional, Union
 from dataclasses import dataclass
-from .dataset import ABFMLDataset
+from abfml.data.dataset import ABFMLDataset
 
 
 @dataclass