fix: po-block works with latest pipelines release

Author: sebffischer

.gitignore

@@ -17,4 +17,5 @@ CRAN-SUBMISSION
 paper/data
 .idea/
 .vsc/
-paper/data
+paper/data
+.vscode/

R/PipeOpTorchBlock.R

@@ -146,7 +146,9 @@ PipeOpTorchBlock = R6Class("PipeOpTorchBlock",
       block = private$.block$clone(deep = TRUE)
       graph = private$.make_graph(block, param_vals$n_blocks)
       inputs = set_names(inputs, graph$input$name)
-      graph$train(inputs, single_input = FALSE)
+      out = graph$train(inputs, single_input = FALSE)
+      self$state = map(out, "pointer_shape")
+      return(out)
     },
     .param_set_base = NULL,
     .additional_phash_input = function() {

tests/testthat/helper_autotest.R

@@ -26,6 +26,7 @@ expect_pipeop_torch = function(graph, id, task, module_class = id, exclude_args
   require_namespaces(c("testthat"))
   po_test = graph$pipeops[[id]]
   result = graph$train(task)
+  expect_true(graph$is_trained)
   md = result[[1]]
 
   # PipeOp overwrites hash and phash if necessary
@@ -321,7 +322,7 @@ expect_torch_callback = function(torch_callback, check_man = TRUE, check_paramse
 
     return(implemented_stages)
   }
-  
+
   implemented_stages = get_all_implemented_stages(cbgen)
   expect_subset(implemented_stages, mlr_reflections$torch$callback_stages)
   expect_true(length(implemented_stages) > 0)

tests/testthat/test_PipeOpTorchBlock.R

@@ -111,3 +111,10 @@ test_that("trafo works", {
   expect_class(network$module_list$`0`$bias, "torch_tensor")
   expect_true(is.null(network$module_list$`1`$bias))
 })
+
+test_that("state", {
+  md = po("torch_ingress_num")$train(list(tsk("iris")))
+  block = nn("block", nn("relu"), n_blocks = 1L)
+  block$train(md)
+  expect_true(block$is_trained)
+})