Delete removed data persistence classes from docs (#1633)

Signed-off-by: Niels Bantilan <[email protected]>
flyteorg · May 12, 2023 · 0a1f289 · 0a1f289
1 parent dd44bba
commit 0a1f289
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 7 deletions.
diff --git a/.github/workflows/docs_build.yml b/.github/workflows/docs_build.yml
@@ -23,4 +23,4 @@ jobs:
         run: |
           sudo apt-get install python3-sphinx
           pip install -r doc-requirements.txt
-          SPHINXOPTS="-W" cd docs && make html
+          cd docs && SPHINXOPTS="-W" make html
diff --git a/flytekit/core/data_persistence.py b/flytekit/core/data_persistence.py
@@ -14,11 +14,7 @@
    :template: custom.rst
    :nosignatures:
 
-   DataPersistence
-   DataPersistencePlugins
-   DiskPersistence
    FileAccessProvider
-   UnsupportedPersistenceOp
 
 """
 import os

diff --git a/flytekit/models/types.py b/flytekit/models/types.py
@@ -259,7 +259,7 @@ def __init__(
         :param flytekit.models.core.types.StructuredDatasetType structured_dataset_type: structured dataset
         :param dict[Text, T] metadata: Additional data describing the type
         :param flytekit.models.annotation.TypeAnnotation annotation: Additional data
-            describing the type _intended to be saturated by the client_
+            describing the type intended to be saturated by the client
         """
         self._simple = simple
         self._schema = schema

diff --git a/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/__init__.py b/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/__init__.py
@@ -8,6 +8,7 @@
    :toctree: generated/
 
    PyTorch
+   Elastic
 """
 
 from .task import Elastic, PyTorch
diff --git a/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py b/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py
@@ -7,7 +7,7 @@
 from typing import Any, Callable, Dict, Optional, Union
 
 import cloudpickle
-from flyteidl.plugins.pytorch_pb2 import DistributedPyTorchTrainingTask, ElasticConfig
+from flyteidl.plugins.pytorch_pb2 import DistributedPyTorchTrainingTask
 from google.protobuf.json_format import MessageToDict
 
 import flytekit
@@ -227,6 +227,8 @@ def get_custom(self, settings: SerializationSettings) -> Optional[Dict[str, Any]
             """
             return super().get_custom(settings)
         else:
+            from flyteidl.plugins.pytorch_pb2 import ElasticConfig
+
             elastic_config = ElasticConfig(
                 rdzv_backend=self.rdzv_backend,
                 min_replicas=self.min_nodes,

diff --git a/plugins/setup.py b/plugins/setup.py
@@ -36,6 +36,7 @@
     "flytekitplugins-onnxpytorch": "flytekit-onnx-pytorch",
     "flytekitplugins-pandera": "flytekit-pandera",
     "flytekitplugins-papermill": "flytekit-papermill",
+    "flytekitplugins-polars": "flytekit-polars",
     "flytekitplugins-ray": "flytekit-ray",
     "flytekitplugins-snowflake": "flytekit-snowflake",
     "flytekitplugins-spark": "flytekit-spark",
-Original file line number
+Diff line change
@@ Expand Up / @@ -8,6 +8,7 @@ @@
        :toctree: generated/
        PyTorch
+       Elastic
     """
     from .task import Elastic, PyTorch