diff --git a/.github/workflows/pythonbuild.yml b/.github/workflows/pythonbuild.yml index 909a26d830..3d54177764 100644 --- a/.github/workflows/pythonbuild.yml +++ b/.github/workflows/pythonbuild.yml @@ -359,7 +359,7 @@ jobs: plugin-names: "flytekit-greatexpectations" # onnxruntime does not support python 3.10 yet # https://github.com/microsoft/onnxruntime/issues/9782 - - python-version: 3.11 + - python-version: 3.12 plugin-names: "flytekit-onnx-pytorch" - python-version: 3.11 plugin-names: "flytekit-onnx-scikitlearn" diff --git a/plugins/flytekit-deck-standard/flytekitplugins/deck/renderer.py b/plugins/flytekit-deck-standard/flytekitplugins/deck/renderer.py index 708e941d88..3ac10ad199 100644 --- a/plugins/flytekit-deck-standard/flytekitplugins/deck/renderer.py +++ b/plugins/flytekit-deck-standard/flytekitplugins/deck/renderer.py @@ -42,9 +42,9 @@ def to_html(self, source_code: str) -> str: Returns: str: The resulting HTML as a string, including CSS and highlighted source code. """ - formatter = pygments.formatters.html.HtmlFormatter(style="colorful") + formatter = pygments.formatters.HtmlFormatter(style="colorful") css = formatter.get_style_defs(".highlight").replace("#fff0f0", "#ffffff") - html = pygments.highlight(source_code, pygments.lexers.python.PythonLexer(), formatter) + html = pygments.highlight(source_code, pygments.lexers.PythonLexer(), formatter) return f"{html}" diff --git a/plugins/flytekit-envd/tests/test_image_spec.py b/plugins/flytekit-envd/tests/test_image_spec.py index b096d80da7..5aa3dc3dab 100644 --- a/plugins/flytekit-envd/tests/test_image_spec.py +++ b/plugins/flytekit-envd/tests/test_image_spec.py @@ -105,11 +105,10 @@ def build(): def test_image_spec_extra_index_url(): image_spec = ImageSpec( - packages=["-U pandas", "torch", "torchvision"], + packages=["torch", "torchvision"], base_image="cr.flyte.org/flyteorg/flytekit:py3.9-latest", pip_extra_index_url=[ "https://download.pytorch.org/whl/cpu", - "https://pypi.anaconda.org/scientific-python-nightly-wheels/simple", ], ) EnvdImageSpecBuilder().build_image(image_spec) @@ -123,10 +122,10 @@ def test_image_spec_extra_index_url(): def build(): base(image="cr.flyte.org/flyteorg/flytekit:py3.9-latest", dev=False) run(commands=[]) - install.python_packages(name=["-U pandas", "torch", "torchvision"]) + install.python_packages(name=["torch", "torchvision"]) install.apt_packages(name=[]) runtime.environ(env={{'PYTHONPATH': '/root:', '_F_IMG_ID': '{image_spec.id}'}}, extra_path=['/root']) - config.pip_index(url="https://pypi.org/simple", extra_url="https://download.pytorch.org/whl/cpu https://pypi.anaconda.org/scientific-python-nightly-wheels/simple") + config.pip_index(url="https://pypi.org/simple", extra_url="https://download.pytorch.org/whl/cpu") """ ) diff --git a/plugins/flytekit-greatexpectations/tests/great_expectations/expectations/test1/demo.json b/plugins/flytekit-greatexpectations/tests/great_expectations/expectations/test1/demo.json index 924916172c..0336cec1c6 100644 --- a/plugins/flytekit-greatexpectations/tests/great_expectations/expectations/test1/demo.json +++ b/plugins/flytekit-greatexpectations/tests/great_expectations/expectations/test1/demo.json @@ -63,23 +63,6 @@ }, "meta": {} }, - { - "expectation_type": "expect_column_values_to_be_in_type_list", - "kwargs": { - "column": "store_and_fwd_flag", - "type_list": [ - "CHAR", - "VARCHAR", - "NVARCHAR", - "TEXT", - "STRING", - "StringType", - "string", - "str" - ] - }, - "meta": {} - }, { "expectation_type": "expect_column_min_to_be_between", "kwargs": { diff --git a/plugins/flytekit-onnx-pytorch/setup.py b/plugins/flytekit-onnx-pytorch/setup.py index cb76eca080..c6d9e714f5 100644 --- a/plugins/flytekit-onnx-pytorch/setup.py +++ b/plugins/flytekit-onnx-pytorch/setup.py @@ -4,7 +4,14 @@ microlib_name = f"flytekitplugins-{PLUGIN_NAME}" -plugin_requires = ["flytekit>=1.3.0b2,<2.0.0", "torch>=1.11.0", "onnx-pytorch", "networkx<3.2; python_version<'3.9'"] +plugin_requires = [ + "flytekit>=1.3.0b2,<2.0.0", + "torch>=1.11.0", + "onnx-pytorch", + "networkx<3.2; python_version<'3.9'", + "onnxscript", + "numpy<2.0.0", +] __version__ = "0.0.0+develop" diff --git a/plugins/flytekit-whylogs/setup.py b/plugins/flytekit-whylogs/setup.py index 0583cbd17d..5385c35f4d 100644 --- a/plugins/flytekit-whylogs/setup.py +++ b/plugins/flytekit-whylogs/setup.py @@ -4,7 +4,9 @@ microlib_name = f"flytekitplugins-{PLUGIN_NAME}" -plugin_requires = ["flytekit>=1.3.0b2", "whylogs[viz]>=1.1.16"] +# whylogs is not compatible with numpy 2.0 yet +# `np.unicode_` was removed in the NumPy 2.0 release. Use `np.str_` instead. +plugin_requires = ["flytekit>=1.3.0b2", "whylogs[viz]>=1.1.16", "numpy<2.0"] __version__ = "0.0.0+develop"