Register default affine method.

mlverse · Apr 14, 2023 · 979ade5 · 979ade5
1 parent e09ffe4
commit 979ade5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -10,6 +10,7 @@ S3method(transform_adjust_hue,default)
 S3method(transform_adjust_hue,torch_tensor)
 S3method(transform_adjust_saturation,default)
 S3method(transform_adjust_saturation,torch_tensor)
+S3method(transform_affine,default)
 S3method(transform_affine,torch_tensor)
 S3method(transform_center_crop,default)
 S3method(transform_color_jitter,default)

diff --git a/R/transforms-defaults.R b/R/transforms-defaults.R
@@ -555,6 +555,7 @@ transform_rotate.default <- function(img, angle, resample = 0, expand = FALSE,
   not_implemented_for_class(img)
 }
 
+#' @export
 transform_affine.default <- function(img, angle, translate, scale, shear,
                                      resample = 0, fillcolor = NULL) {
   not_implemented_for_class(img)