embiggen/edge_prediction/edge_prediction_tensorflow/__init__.py
"""Submodule providing edge prediction models based on TensorFlow Models."""
from embiggen.edge_prediction.edge_prediction_model import AbstractEdgePredictionModel
from embiggen.utils.abstract_models import build_init
build_init(
module_library_names="tensorflow",
formatted_library_name="TensorFlow",
task_name="Edge Prediction",
expected_parent_class=AbstractEdgePredictionModel
)