hugging_face_client/api/
get_parquet.rs

1use std::collections::HashMap;
2
3use serde::{Deserialize, Serialize};
4
5/// Request of [`crate::client::Client::get_parquet`]
6#[derive(Debug, Serialize)]
7pub struct GetParquetReq<'a> {
8  #[serde(rename = "repo_id")]
9  pub(crate) repo_name: &'a str,
10
11  pub(crate) subset: Option<&'a str>,
12
13  pub(crate) split: Option<&'a str>,
14}
15
16impl<'a> GetParquetReq<'a> {
17  pub fn new(repo_name: &'a str) -> GetParquetReq<'a> {
18    GetParquetReq {
19      repo_name,
20      subset: None,
21      split: None,
22    }
23  }
24
25  pub fn subset(mut self, subset: &'a str) -> Self {
26    self.subset = Some(subset);
27    self
28  }
29
30  pub fn split(mut self, split: &'a str) -> Self {
31    self.split = Some(split);
32    self
33  }
34}
35
36/// Response of [`crate::client::Client::get_parquet`]
37#[derive(Debug, Deserialize)]
38#[serde(untagged)]
39pub enum GetParquetRes {
40  Repo(HashMap<String, HashMap<String, Vec<String>>>),
41  Subset(HashMap<String, Vec<String>>),
42  Split(Vec<String>),
43}