lynxkite-graph-analytics/src/lynxkite_graph_analytics/__init__.py

"""Graph analytics environment for LynxKite. The core types and functions are imported here for easy access."""

import os
import pandas as pd

if os.environ.get("NX_CUGRAPH_AUTOCONFIG", "").strip().lower() == "true":
    import cudf.pandas

    cudf.pandas.install()

pd.options.mode.copy_on_write = True  # Prepare for Pandas 3.0.

from .core import *  # noqa (easier access for core classes)
from . import lynxkite_ops  # noqa (imported to trigger registration)
from . import networkx_ops  # noqa (imported to trigger registration)
from . import pytorch_model_ops  # noqa (imported to trigger registration)

if os.environ.get("LYNXKITE_BIONEMO_INSTALLED", "").strip().lower() == "true":
    from . import bionemo_ops  # noqa (imported to trigger registration)