deployTFModel and writeManifest support TensorFlow

NEWS.md

@@ -9,6 +9,9 @@
 * `deployApp()` and `writeManifest()` receive the default value for the
   `image` argument from the `RSCONNECT_IMAGE` environment variable. (#1063)
 
+* `deployTF()` can deploy a TensorFlow model to Posit Connect. Requires Posit
+  Connect 2024.05.0 or higher.
+
 # rsconnect 1.2.2
 
 * Use internally computed SHA1 sums and PKI signing when SHA1 is disabled 

R/appDependencies.R

@@ -136,7 +136,7 @@ appDependencies <- function(appDir = getwd(),
 }
 
 needsR <- function(appMetadata) {
-  if (appMetadata$appMode == "static") {
+  if (appMetadata$appMode %in% c("static", "tensorflow-saved-model")) {
     return(FALSE)
   }
 

R/appMetadata.R

@@ -38,9 +38,9 @@ appMetadata <- function(appDir,
       appMode <- "shiny"
     } else {
       # Inference only uses top-level files
-      rootFiles <- appFiles[dirname(appFiles) == "."]
       appMode <- inferAppMode(
-        file.path(appDir, rootFiles),
+        appDir,
+        appFiles,
         usesQuarto = quarto,
         isShinyappsServer = isShinyappsServer
       )
@@ -98,10 +98,18 @@ checkAppLayout <- function(appDir, appPrimaryDoc = NULL) {
   # can be run/served.
 }
 
-# infer the mode of the application from files in the root dir
-inferAppMode <- function(absoluteRootFiles,
-                         usesQuarto = NA,
-                         isShinyappsServer = FALSE) {
+# Infer the mode of the application from included files. Most content types
+# only consider files at the directory root. TensorFlow saved models may be
+# anywhere in the hierarchy.
+inferAppMode <- function(
+  appDir,
+  appFiles,
+  usesQuarto = NA,
+  isShinyappsServer = FALSE) {
+
+  rootFiles <- appFiles[dirname(appFiles) == "."]
+  absoluteRootFiles <- file.path(appDir, rootFiles)
+  absoluteAppFiles <- file.path(appDir, appFiles)
 
   matchingNames <- function(paths, pattern) {
     idx <- grepl(pattern, basename(paths), ignore.case = TRUE, perl = TRUE)
@@ -190,6 +198,12 @@ inferAppMode <- function(absoluteRootFiles,
     return("quarto-static")
   }
 
+  # TensorFlow model files are lower in the hierarchy, not at the root.
+  modelFiles <- matchingNames(absoluteAppFiles, "^(saved_model.pb|saved_model.pbtxt)$")
+  if (length(modelFiles) > 0) {
+    return("tensorflow-saved-model")
+  }
+
   # no renderable content
   "static"
 }

R/deployTFModel.R

@@ -1,20 +1,11 @@
 #' Deploy a TensorFlow saved model
 #'
-#' This function is defunct. Posit Connect no longer supports hosting of
-#' TensorFlow Model APIs. A TensorFlow model can be deployed as a
-#' [Plumber API](https://tensorflow.rstudio.com/guides/deploy/plumber.html),
-#' [Shiny application](https://tensorflow.rstudio.com/guides/deploy/shiny), or
-#' other supported content type.
+#' Deploys a directory containing a TensorFlow saved model.
 #'
-#' @param modelDir Path to the saved model directory. Unused.
-#' @param ... Unused.
+#' #' @inheritParams deployApp
 #'
 #' @family Deployment functions
 #' @export
-deployTFModel <- function(modelDir, ...) {
-  lifecycle::deprecate_stop(
-    when = "1.0.0",
-    what = "deployTFModel()",
-    details = "Posit Connect no longer supports hosting of TensorFlow Model APIs."
-  )
+deployTFModel <- function(...) {
+  deployApp(appMode = "tensorflow-saved-model", ...)
 }

tests/testthat/test-writeManifest.R

@@ -239,6 +239,18 @@ test_that("environment.image is not set when image is not provided", {
   expect_null(manifest$environment)
 })
 
+test_that("TensorFlow models are identified", {
+  skip_on_cran()
+
+  app_dir <- local_temp_app(list(
+    "1/saved_model.pb" = "fake-saved-model"
+  ))
+  manifest <- makeManifest(app_dir)
+  expect_equal(manifest$metadata$appmode, "tensorflow-saved-model")
+  expect_null(manifest$packages)
+  expect_named(manifest$files, c("1/saved_model.pb"))
+})
+
 test_that("environment.image is set when image is provided", {
   skip_on_cran()