pytyche.setup

Runtime setup inspection (pt.check_setup) and the no-CUDA fit warning.

JAX is imported inside functions, never at module level — import pytyche must stay importable without pulling the heavy stack (see pytyche/__init__).

Functions

check_setup()

Inspect the pytyche runtime and report on it.

Classes

SetupReport(pytyche_version, jax_devices, ...)

Snapshot of the pytyche runtime, returned by check_setup().

Exceptions

NoCudaWarning

Emitted once per process on the first fit without a CUDA device.

exception pytyche.setup.NoCudaWarning

Bases: UserWarning

Emitted once per process on the first fit without a CUDA device.

class pytyche.setup.SetupReport(pytyche_version, jax_devices, cuda_available, bartz_version, calibration_registry, recommended_install)

Bases: object

Snapshot of the pytyche runtime, returned by check_setup().

pytyche_version

Installed pytyche version.

jax_devices

String form of every device JAX reports.

cuda_available

Whether any reported device is a GPU.

bartz_version

Installed bartz version.

calibration_registry

Names of calibration artifacts bundled with the wheel (empty when none are bundled).

recommended_install

Install command closing a detected setup gap (e.g. the CUDA upgrade), or None when the setup is complete.

Parameters:

• pytyche_version (str)

• jax_devices (list[str])

• cuda_available (bool)

• bartz_version (str)

• calibration_registry (list[str])

• recommended_install (str | None)

pytyche.setup.check_setup()

Inspect the pytyche runtime and report on it.

Prints a human-readable summary to stdout (for notebook / interactive use) and returns the same content as a structured SetupReport (for programmatic checks). Idempotent — no side effects beyond the print; in particular it never triggers the no-CUDA fit warning.

Return type:

SetupReport

Returns:

The SetupReport for this process.