Source code for gempy.core.data.gempy_engine_config

from dataclasses import dataclass
from typing import Optional
import os

from gempy_engine import config
from gempy_engine.config import AvailableBackends




[docs]
@dataclass
class GemPyEngineConfig:
    backend: AvailableBackends = config.DEFAULT_BACKEND # ? This can be grabbed from gempy.config file?
    use_gpu: bool = os.getenv("GEMPY_USE_GPU", "False") == "True"
    dtype: Optional[str] = None  #: The data type used in the engine. If None, the default data type of the backend is used.
    compute_grads: Optional[bool] = False