Flytekit OmegaConf Plugin
Flytekit python natively supports serialization of many data types for exchanging information between tasks.
The Flytekit OmegaConf Plugin extends these by the DictConfig type from the
OmegaConf package as well as related types
that are being used by the hydra package for configuration management.
Task example
from dataclasses import dataclass
import flytekitplugins.omegaconf # noqa F401
from flytekit import task, workflow
from omegaconf import DictConfig
@dataclass
class MySimpleConf:
_target_: str = "lightning_module.MyEncoderModule"
learning_rate: float = 0.0001
@task
def my_task(cfg: DictConfig) -> None:
print(f"Doing things with {cfg.learning_rate=}")
@workflow
def pipeline(cfg: DictConfig) -> None:
my_task(cfg=cfg)
if __name__ == "__main__":
from omegaconf import OmegaConf
cfg = OmegaConf.structured(MySimpleConf)
pipeline(cfg=cfg)
Transformer configuration
The transformer can be set to one of three modes:
Dataclass - This mode should be used with a StructuredConfig and will reconstruct the config from the matching dataclass
during deserialisation in order to make typing information from the dataclass and continued validation thereof available.
This requires the dataclass definition to be available via python import in the Flyte execution environment in which
objects are (de-)serialised.
DictConfig - This mode will deserialize the config into a DictConfig object. In particular, dataclasses are translated
into DictConfig objects and only primitive types are being checked. The definition of underlying dataclasses for
structured configs is only required during the initial serialization for this mode.
Auto - This mode will try to deserialize according to the Dataclass mode and fall back to the DictConfig mode if the
dataclass definition is not available. This is the default mode.
You can set the transformer mode globally or for the current context only the following ways:
from flytekitplugins.omegaconf import set_transformer_mode, set_local_transformer_mode, OmegaConfTransformerMode
# Set the global transformer mode using the new function
set_transformer_mode(OmegaConfTransformerMode.DictConfig)
# You can also the mode for the current context only
with set_local_transformer_mode(OmegaConfTransformerMode.Dataclass):
# This will use the Dataclass mode
pass
Since the DictConfig is flattened and keys transformed into dot notation, the keys of the DictConfig must not contain
dots.
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file flytekitplugins_omegaconf-1.16.27.tar.gz.
File metadata
- Download URL: flytekitplugins_omegaconf-1.16.27.tar.gz
- Upload date:
- Size: 12.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/7.0.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0a7a4e45f0b0e5afa7157042685a2fa9f2b74d227622d109e6c800bf65e9d5a8
|
|
| MD5 |
36d115636493038bf56c6f1db0e9c6f0
|
|
| BLAKE2b-256 |
60616cbced31306cbac30a8a7ee780b740d8735b6fcc81716558c5777739deae
|
File details
Details for the file flytekitplugins_omegaconf-1.16.27-py3-none-any.whl.
File metadata
- Download URL: flytekitplugins_omegaconf-1.16.27-py3-none-any.whl
- Upload date:
- Size: 10.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/7.0.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c12b5bd96f9bfbda50655c05bc10fbd5aff1b319f20b33eeb8b382bb8fc1ea19
|
|
| MD5 |
cba8e169cfcd96a40f409fe1ee999123
|
|
| BLAKE2b-256 |
aa2afe3b8b00b3540e4524e95cc30a499cf4509767eb53cbb3dfdeb7a1241545
|
