File size: 763 Bytes
44a2d59
 
 
 
 
 
 
 
 
 
 
 
dbf89c5
 
fc43558
 
7baf2a1
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""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)