Skip to content
Merged
2 changes: 1 addition & 1 deletion .github/workflows/pythonbuild.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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"<style>{css}</style>{html}"


Expand Down
7 changes: 3 additions & 4 deletions plugins/flytekit-envd/tests/test_image_spec.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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")
"""
)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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": {
Expand Down
9 changes: 8 additions & 1 deletion plugins/flytekit-onnx-pytorch/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down
4 changes: 3 additions & 1 deletion plugins/flytekit-whylogs/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"

Expand Down
Loading