R/class_torch.R

In targets: Dynamic Function-Oriented 'Make'-Like Declarative Pipelines

#' @export
store_class_format.torch <- function(format) {
  c("tar_torch", "tar_nonexportable", "tar_store")
}

#' @export
store_assert_format_setting.torch <- function(format) {
}

#' @export
store_read_path.tar_torch <- function(store, path) {
  torch::torch_load(path)
}

#' @export
store_write_path.tar_torch <- function(store, object, path) {
  torch::torch_save(obj = object, path = path)
}

#' @export
store_marshal_object.tar_torch <- function(store, object) {
  con <- rawConnection(raw(), open = "wr")
  on.exit(close(con))
  torch::torch_save(object, con)
  rawConnectionValue(con)
}

#' @export
store_unmarshal_object.tar_torch <- function(store, object) {
  con <- rawConnection(object, open = "r")
  on.exit(close(con))
  torch::torch_load(con)
}

#' @export
store_get_packages.tar_torch <- function(store) {
  "torch"
}