relayrl_types 0.6.0

Data types for the RelayRL framework.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

pub mod data;
#[cfg(all(
    any(feature = "tch-model", feature = "onnx-model"),
    any(feature = "ndarray-backend", feature = "tch-backend")
))]
pub mod model;

pub mod prelude {
    #[cfg(any(feature = "ndarray-backend", feature = "tch-backend"))]
    pub mod action {
        pub use crate::data::action::{
            ActionError, CodecConfig, EncodedAction, RelayRLAction, RelayRLData,
        };
    }

    #[cfg(any(feature = "ndarray-backend", feature = "tch-backend"))]
    pub mod tensor {
        pub mod relayrl {
            pub use crate::data::tensor::{
                AnyBurnTensor, BackendMatcher, BoolBurnTensor, DType, DeviceType, FloatBurnTensor,
                IntBurnTensor, SupportedTensorBackend, TensorData, TensorError,
            };
        }

        pub mod burn {
            pub use burn_tensor::*;
        }
    }

    #[cfg(any(feature = "ndarray-backend", feature = "tch-backend"))]
    pub mod trajectory {
        pub use crate::data::trajectory::{
            EncodedTrajectory, RelayRLTrajectory, RelayRLTrajectoryTrait, TrajectoryError,
        };
    }

    #[cfg(any(feature = "ndarray-backend", feature = "tch-backend"))]
    pub mod records {
        pub use crate::data::records::arrow::{ArrowTrajectory, ArrowTrajectoryError};
        pub use crate::data::records::csv::{CsvTrajectory, CsvTrajectoryError};
    }

    #[cfg(all(
        any(feature = "tch-model", feature = "onnx-model"),
        any(feature = "ndarray-backend", feature = "tch-backend")
    ))]
    pub mod model {
        pub use crate::model::{HotReloadableModel, ModelError, ModelModule};
    }

    pub mod codec {
        #[cfg(feature = "compression")]
        pub use crate::data::utilities::compress::{CompressedData, CompressionScheme};

        #[cfg(feature = "integrity")]
        pub use crate::data::utilities::integrity::{VerifiedData, compute_checksum};

        #[cfg(feature = "encryption")]
        pub use crate::data::utilities::encrypt::{EncryptedData, EncryptionKey};

        #[cfg(feature = "metadata")]
        pub use crate::data::utilities::metadata::TensorMetadata;

        #[cfg(feature = "quantization")]
        pub use crate::data::utilities::quantize::{QuantizationScheme, QuantizedData};

        #[cfg(feature = "integrity")]
        pub use crate::data::utilities::chunking::{ChunkedTensor, TensorChunk};
    }
}

use std::collections::HashMap;

/// Hyperparams enum represents hyperparameter inputs
#[derive(Clone, Debug, serde::Serialize, serde::Deserialize)]
pub enum HyperparameterArgs {
    Map(HashMap<String, String>),
    List(Vec<String>),
}