Source code for gigl.src.common.utils.data.feature_serialization

from typing import Sequence

import numpy as np

from gigl.common.utils.compute.serialization.serialize_np import NumpyCoder



[docs]
class FeatureSerializationUtils:

[docs]
    coder = NumpyCoder()


    @classmethod

[docs]
    def deserialize_node_features(
        cls, serialized_features: Sequence[float]
    ) -> Sequence[float]:
        return serialized_features


    @classmethod

[docs]
    def serialize_node_features(cls, features: np.ndarray) -> np.ndarray:
        return features


    @classmethod

[docs]
    def deserialize_edge_features(
        cls, serialized_features: Sequence[float]
    ) -> Sequence[float]:
        return serialized_features


    @classmethod

[docs]
    def serialize_edge_features(cls, features: np.ndarray) -> np.ndarray:
        return features