liboxen 0.50.6

Oxen is a fast, unstructured data version control, to help version large machine learning datasets written in Rust.
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

use strum::VariantNames;

use crate::{
    error::OxenError,
    model::{Schema, staged_row_status::StagedRowStatus},
};

pub mod changes_db;
pub mod column_changes_db;
pub mod columns;
pub mod df_db;
pub mod row_changes_db;
pub mod rows;
pub mod workspace_df_db;

#[derive(Debug, thiserror::Error)]
pub enum DataFrameError {
    #[error("df must have exactly one row to be used for modification")]
    ModifyOnly1Row,

    /// A column name was requested in a dataframe, but no such column exists.
    #[error("Column name not found: {0}")]
    ColumnNameNotFound(String),

    /// A column name already exists in the dataframe's schema and cannot be added again.
    #[error("Column name already exists: {0}")]
    ColumnNameAlreadyExists(String),

    #[error("Failed to create df db directory: {0}")]
    FailCreateDfDbDir(std::io::Error),

    #[error("Failed to open df db: {0}")]
    FailOpenDfDb(Box<Self>),

    #[error("No rows in table {0}")]
    NoRowsInTable(String),

    #[error("Invalid file type: expected .csv, .tsv, .parquet, .jsonl, .json, .ndjson")]
    InvalidFileType,

    #[error("modify_row incompatible_schemas {table_schema:?}\n{df_cols:?}")]
    IncompatibleSchemas {
        table_schema: Schema,
        df_cols: Vec<String>,
    },

    #[error("Diff status column is not a string")]
    DiffStatusColNotStr,

    #[error("Missing diff status column")]
    MissingDiffStatusCol,

    #[error("Diff hash column is not a string")]
    DiffHashColNotStr,

    #[error("Expected {expected} rows to be modified, but got {actual}")]
    UnexpectedModifications { expected: usize, actual: usize },

    #[error("Invalid row status: \"{0}\". Expecting one of: {valid}", valid=<StagedRowStatus as VariantNames>::VARIANTS.join(", "))]
    InvalidRowStatus(String),

    #[error("Row status not found")]
    RowStatusNotFound,

    #[error("DataFrame with UUID {0} not found.")]
    MissingDataFrame(String),

    #[error("Must index embeddings before querying")]
    MustIndexEmbeddings,

    #[error("All embeddings must be the same length")]
    EmbeddingLengthMismatch,

    #[error("Expected Float32Array inside ListArray")]
    ExpectedF32ArrayInside,

    #[error("Expected arrow::datatypes::DataType::Float32 inside List")]
    ExpectedF32,

    #[error("Failed to downcast to FixedSizeListArray")]
    FailFixedSizeDowncast,

    #[error("Column FixedSizeList must be a float32 type")]
    ExpectF32FixedSizeList,

    #[error("Expected arrow::datatypes::DataType::List inside as data type")]
    ExpectListInside,

    #[error("Failed to downcast to ListArray")]
    FailListDowncast,

    #[error("Expected Float64Array inside ListArray")]
    ExpectF64ArrayInside,

    #[error("Column must be a list of float32 or float64")]
    ExpectColFloats,

    #[error("Column must be a list type")]
    ExpectList,

    /// No rows were found for a given SQL query.
    #[error("Query returned no rows")]
    NoRowsFound,

    #[error("Vectors must have a length greater than 0")]
    EmptyEmbedding,

    #[error("Column does not exist in embedding configuration: {0}")]
    ColNotFoundInConfig(String),

    #[error("Failed to read embedding configuration file: {0}")]
    FailReadConfig(Box<OxenError>), // TODO: change to FsError when PR lands

    #[error("Failed to write embedding configuration: {0}")]
    FailWriteConfig(std::io::Error),

    #[error("No SELECT found in query")]
    NoSelectInQuery,

    #[error("No FROM found in query")]
    NoFromInQuery,

    #[error("Dataset is not indexed")]
    NotIndexed,

    #[error("Could not create parent directory for DuckDB file: {0}")]
    CreateParent(Box<OxenError>), // TODO: change to FsError when PR lands

    // #[error("Resource not found: {0}")]
    // ResourceNotFound(String),
    #[error("{0}")]
    DuckDb(#[from] duckdb::Error),

    #[error("{0}")]
    Arrow(#[from] arrow::error::ArrowError),

    #[error("{0}")]
    Polars(#[from] polars::error::PolarsError),

    #[error("Failed to parse SQL: {0}")]
    SqlParse(#[from] sqlparser::parser::ParserError),

    #[error("{0}")]
    RocksDb(#[from] rocksdb::Error),

    #[error("{0}")]
    SerdeJson(#[from] serde_json::error::Error),

    #[error("{0}")]
    TomlDe(#[from] toml::de::Error),

    #[error("{0}")]
    TomlSer(#[from] toml::ser::Error),
}