Skip to content

utils.io

IO utils for AtomicEmbeddings.

NumpyEncoder

Bases: JSONEncoder

Special json encoder for numpy types.

Source code in src/elementembeddings/utils/io.py 
11
12
13
14
15
16
17
18
class NumpyEncoder(json.JSONEncoder):
    """Special json encoder for numpy types."""

    def default(self, o: Any) -> Any:
        """Encode numpy types."""
        if isinstance(o, np.ndarray):
            return o.tolist()
        return super().default(o)

default(o)

Encode numpy types.

Source code in src/elementembeddings/utils/io.py 
14
15
16
17
18
def default(self, o: Any) -> Any:
    """Encode numpy types."""
    if isinstance(o, np.ndarray):
        return o.tolist()
    return super().default(o)