huggingface

type Client interface {
	GetDatasetSize(ctx context.Context) (*DatasetSizeResponse, error)
	GetDatasetRows(ctx context.Context, offset, length int) (*RowResponse, error)
	GetDatasetInfo(ctx context.Context) (*DatasetInfoResponse, error)
}

type ClientImpl struct {
	// contains filtered or unexported fields
}

type ClientMock struct {
	// GetDatasetInfoFunc mocks the GetDatasetInfo method.
	GetDatasetInfoFunc func(ctx context.Context) (*DatasetInfoResponse, error)

	// GetDatasetRowsFunc mocks the GetDatasetRows method.
	GetDatasetRowsFunc func(ctx context.Context, offset int, length int) (*RowResponse, error)

	// GetDatasetSizeFunc mocks the GetDatasetSize method.
	GetDatasetSizeFunc func(ctx context.Context) (*DatasetSizeResponse, error)
	// contains filtered or unexported fields
}

type DatasetInfoResponse struct {
	Features map[string]any `json:"features"`
}

type DatasetSizeResponse struct {
	Size SizeInfo `json:"size"`
}

type RowInfo struct {
	Row map[string]any `json:"row"`
}

type RowResponse struct {
	Rows []RowInfo `json:"rows"`
}

type SizeInfo struct {
	Splits []SplitInfo `json:"splits"`
}

type SnapshotTransport struct {
	BaseTransport http.RoundTripper
}

type SplitInfo struct {
	NumRows int    `json:"num_rows"`
	Config  string `json:"config"`
	Split   string `json:"split"`
}