Skip to content

Commit 75dd006

Browse files
authored
Fixing the azure integration and marshmallow (#3626)
* fixing the azure integration and marshmallow * linting * new linting
1 parent ec72521 commit 75dd006

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

src/zenml/integrations/azure/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,9 @@ class AzureIntegration(Integration):
5353
# an attribute that was removed in Numpy 2.0. However, AzureML itself
5454
# does not have a limitation on numpy.
5555
"numpy<2.0",
56+
# Marshmallow>4.0 leads to the following error with the AzureML SDK:
57+
# ImportError: cannot import name 'FieldInstanceResolutionError' from 'marshmallow.utils'
58+
"marshmallow<4.0.0",
5659
]
5760
REQUIREMENTS_IGNORED_ON_UNINSTALL = ["kubernetes", "numpy"]
5861

src/zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -883,7 +883,7 @@ def _get_session_cookie(self, username: str, password: str) -> str:
883883
f"Error while trying to fetch kubeflow cookie: {errh}"
884884
)
885885

886-
cookie_dict: Dict[str, str] = session.cookies.get_dict() # type: ignore[no-untyped-call]
886+
cookie_dict: Dict[str, str] = session.cookies.get_dict() # type: ignore[no-untyped-call, unused-ignore]
887887

888888
if "authservice_session" not in cookie_dict:
889889
raise RuntimeError("Invalid username and/or password!")

src/zenml/integrations/tekton/orchestrators/tekton_orchestrator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def _get_session_cookie(self, username: str, password: str) -> str:
243243
raise RuntimeError(
244244
f"Error while trying to fetch tekoton cookie: {errh}"
245245
)
246-
cookie_dict: Dict[str, str] = session.cookies.get_dict() # type: ignore[no-untyped-call]
246+
cookie_dict: Dict[str, str] = session.cookies.get_dict() # type: ignore[no-untyped-call, unused-ignore]
247247

248248
if "authservice_session" not in cookie_dict:
249249
raise RuntimeError("Invalid username and/or password!")

0 commit comments

Comments
 (0)