docs: add docs for tensorflow type

Signed-off-by: Tushar Mittal <[email protected]>
flyteorg · Oct 18, 2022 · 42d3362 · 42d3362
1 parent 28c8227
commit 42d3362
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/docs/source/extras.tensorflow.rst b/docs/source/extras.tensorflow.rst
@@ -0,0 +1,7 @@
+############
+Tensorflow Type
+############
+.. automodule:: flytekit.extras.tensorflow
+   :no-members:
+   :no-inherited-members:
+   :no-special-members:
diff --git a/docs/source/types.extend.rst b/docs/source/types.extend.rst
@@ -12,3 +12,4 @@ Feel free to follow the pattern of the built-in types.
    types.builtins.file
    types.builtins.directory
    extras.pytorch
+   extras.tensorflow