add weightwatchers plugin

plugins/flytekit-deck-weightwatcher/README.md

@@ -0,0 +1,9 @@
+# Flytekit Deck Plugin
+
+This Plugin provides more renderers to improve task visibility.
+
+To install the plugin, run the following command:
+
+```bash
+pip install flytekitplugins-deck-standard
+```

plugins/flytekit-deck-weightwatcher/flytekitplugins/deck_weightwatcher/__init__.py

@@ -0,0 +1,15 @@
+"""
+.. currentmodule:: flytekitplugins.deck
+
+This package contains things that are useful when extending Flytekit.
+
+.. autosummary::
+   :template: custom.rst
+   :toctree: generated/
+
+   BoxRenderer
+   FrameProfilingRenderer
+   MarkdownRenderer
+"""
+
+from .renderer import BoxRenderer, WeightWatcherRenderer

plugins/flytekit-deck-weightwatcher/flytekitplugins/deck_weightwatcher/renderer.py

@@ -0,0 +1,8 @@
+import weightwatcher as ww
+
+
+class WeightWatcherRenderer:
+    def to_html(self, model) -> str:
+        watcher = ww.WeightWatcher(model=model)
+        details = watcher.analyze()
+        return df.to_html()

plugins/flytekit-deck-weightwatcher/requirements.in

@@ -0,0 +1,2 @@
+.
+-e file:.#egg=flytekitplugins-deck-standard

plugins/flytekit-deck-weightwatcher/requirements.txt

@@ -0,0 +1,336 @@
+#
+# This file is autogenerated by pip-compile with Python 3.11
+# by the following command:
+#
+#    pip-compile requirements.in
+#
+-e file:.#egg=flytekitplugins-deck-standard
+    # via -r requirements.in
+appnope==0.1.3
+    # via ipython
+arrow==1.2.3
+    # via jinja2-time
+asttokens==2.2.1
+    # via stack-data
+attrs==22.2.0
+    # via visions
+backcall==0.2.0
+    # via ipython
+binaryornot==0.4.4
+    # via cookiecutter
+certifi==2022.12.7
+    # via requests
+cffi==1.15.1
+    # via cryptography
+chardet==5.1.0
+    # via binaryornot
+charset-normalizer==3.1.0
+    # via requests
+click==8.1.3
+    # via
+    #   cookiecutter
+    #   flytekit
+cloudpickle==2.2.1
+    # via flytekit
+contourpy==1.0.7
+    # via matplotlib
+cookiecutter==2.1.1
+    # via flytekit
+croniter==1.3.8
+    # via flytekit
+cryptography==39.0.2
+    # via pyopenssl
+cycler==0.11.0
+    # via matplotlib
+dataclasses-json==0.5.7
+    # via flytekit
+decorator==5.1.1
+    # via
+    #   ipython
+    #   retry
+deprecated==1.2.13
+    # via flytekit
+diskcache==5.4.0
+    # via flytekit
+docker==6.0.1
+    # via flytekit
+docker-image-py==0.1.12
+    # via flytekit
+docstring-parser==0.15
+    # via flytekit
+executing==1.2.0
+    # via stack-data
+flyteidl==1.3.14
+    # via flytekit
+flytekit==1.3.1
+    # via flytekitplugins-deck-standard
+fonttools==4.39.2
+    # via matplotlib
+gitdb==4.0.10
+    # via gitpython
+gitpython==3.1.31
+    # via flytekit
+googleapis-common-protos==1.59.0
+    # via
+    #   flyteidl
+    #   flytekit
+    #   grpcio-status
+grpcio==1.51.3
+    # via
+    #   flytekit
+    #   grpcio-status
+grpcio-status==1.51.3
+    # via flytekit
+htmlmin==0.1.12
+    # via ydata-profiling
+idna==3.4
+    # via requests
+imagehash==4.3.1
+    # via
+    #   visions
+    #   ydata-profiling
+importlib-metadata==6.1.0
+    # via
+    #   flytekit
+    #   keyring
+ipython==8.11.0
+    # via ipywidgets
+ipywidgets==8.0.5
+    # via flytekitplugins-deck-standard
+jaraco-classes==3.2.3
+    # via keyring
+jedi==0.18.2
+    # via ipython
+jinja2==3.1.2
+    # via
+    #   cookiecutter
+    #   jinja2-time
+    #   ydata-profiling
+jinja2-time==0.2.0
+    # via cookiecutter
+joblib==1.2.0
+    # via
+    #   flytekit
+    #   phik
+jupyterlab-widgets==3.0.6
+    # via ipywidgets
+keyring==23.13.1
+    # via flytekit
+kiwisolver==1.4.4
+    # via matplotlib
+markdown==3.4.2
+    # via flytekitplugins-deck-standard
+markupsafe==2.1.2
+    # via jinja2
+marshmallow==3.19.0
+    # via
+    #   dataclasses-json
+    #   marshmallow-enum
+    #   marshmallow-jsonschema
+marshmallow-enum==1.5.1
+    # via dataclasses-json
+marshmallow-jsonschema==0.13.0
+    # via flytekit
+matplotlib==3.6.3
+    # via
+    #   phik
+    #   seaborn
+    #   ydata-profiling
+matplotlib-inline==0.1.6
+    # via ipython
+more-itertools==9.1.0
+    # via jaraco-classes
+multimethod==1.9.1
+    # via
+    #   visions
+    #   ydata-profiling
+mypy-extensions==1.0.0
+    # via typing-inspect
+natsort==8.3.1
+    # via flytekit
+networkx==3.0
+    # via visions
+numpy==1.23.5
+    # via
+    #   contourpy
+    #   flytekit
+    #   imagehash
+    #   matplotlib
+    #   pandas
+    #   patsy
+    #   phik
+    #   pyarrow
+    #   pywavelets
+    #   scipy
+    #   seaborn
+    #   statsmodels
+    #   visions
+    #   ydata-profiling
+packaging==23.0
+    # via
+    #   docker
+    #   marshmallow
+    #   matplotlib
+    #   statsmodels
+pandas==1.5.3
+    # via
+    #   flytekit
+    #   phik
+    #   seaborn
+    #   statsmodels
+    #   visions
+    #   ydata-profiling
+parso==0.8.3
+    # via jedi
+patsy==0.5.3
+    # via statsmodels
+pexpect==4.8.0
+    # via ipython
+phik==0.12.3
+    # via ydata-profiling
+pickleshare==0.7.5
+    # via ipython
+pillow==9.4.0
+    # via
+    #   imagehash
+    #   matplotlib
+    #   visions
+plotly==5.13.1
+    # via flytekitplugins-deck-standard
+prompt-toolkit==3.0.38
+    # via ipython
+protobuf==4.22.1
+    # via
+    #   flyteidl
+    #   googleapis-common-protos
+    #   grpcio-status
+    #   protoc-gen-swagger
+protoc-gen-swagger==0.1.0
+    # via flyteidl
+ptyprocess==0.7.0
+    # via pexpect
+pure-eval==0.2.2
+    # via stack-data
+py==1.11.0
+    # via retry
+pyarrow==10.0.1
+    # via flytekit
+pycparser==2.21
+    # via cffi
+pydantic==1.10.6
+    # via ydata-profiling
+pygments==2.14.0
+    # via ipython
+pyopenssl==23.0.0
+    # via flytekit
+pyparsing==3.0.9
+    # via matplotlib
+python-dateutil==2.8.2
+    # via
+    #   arrow
+    #   croniter
+    #   flytekit
+    #   matplotlib
+    #   pandas
+python-json-logger==2.0.7
+    # via flytekit
+python-slugify==8.0.1
+    # via cookiecutter
+pytimeparse==1.1.8
+    # via flytekit
+pytz==2022.7.1
+    # via
+    #   flytekit
+    #   pandas
+pywavelets==1.4.1
+    # via imagehash
+pyyaml==6.0
+    # via
+    #   cookiecutter
+    #   flytekit
+    #   responses
+    #   ydata-profiling
+regex==2022.10.31
+    # via docker-image-py
+requests==2.28.2
+    # via
+    #   cookiecutter
+    #   docker
+    #   flytekit
+    #   responses
+    #   ydata-profiling
+responses==0.23.1
+    # via flytekit
+retry==0.9.2
+    # via flytekit
+scipy==1.9.3
+    # via
+    #   imagehash
+    #   phik
+    #   statsmodels
+    #   ydata-profiling
+seaborn==0.12.2
+    # via ydata-profiling
+six==1.16.0
+    # via
+    #   asttokens
+    #   patsy
+    #   python-dateutil
+smmap==5.0.0
+    # via gitdb
+sortedcontainers==2.4.0
+    # via flytekit
+stack-data==0.6.2
+    # via ipython
+statsd==3.3.0
+    # via flytekit
+statsmodels==0.13.5
+    # via ydata-profiling
+tangled-up-in-unicode==0.2.0
+    # via visions
+tenacity==8.2.2
+    # via plotly
+text-unidecode==1.3
+    # via python-slugify
+tqdm==4.64.1
+    # via ydata-profiling
+traitlets==5.9.0
+    # via
+    #   ipython
+    #   ipywidgets
+    #   matplotlib-inline
+typeguard==2.13.3
+    # via ydata-profiling
+types-pyyaml==6.0.12.8
+    # via responses
+typing-extensions==4.5.0
+    # via
+    #   flytekit
+    #   pydantic
+    #   typing-inspect
+typing-inspect==0.8.0
+    # via dataclasses-json
+urllib3==1.26.15
+    # via
+    #   docker
+    #   flytekit
+    #   requests
+    #   responses
+visions[type_image_path]==0.7.5
+    # via ydata-profiling
+wcwidth==0.2.6
+    # via prompt-toolkit
+websocket-client==1.5.1
+    # via docker
+wheel==0.40.0
+    # via flytekit
+widgetsnbextension==4.0.6
+    # via ipywidgets
+wrapt==1.15.0
+    # via
+    #   deprecated
+    #   flytekit
+ydata-profiling==4.1.1
+    # via flytekitplugins-deck-standard
+zipp==3.15.0
+    # via importlib-metadata