From 71da307f2badb133b2b214adb0e67736c9c7a61b Mon Sep 17 00:00:00 2001 From: Baris Can Durak Date: Mon, 23 Sep 2024 16:57:26 +0200 Subject: [PATCH 1/2] unpin numpy version --- src/zenml/integrations/numpy/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/zenml/integrations/numpy/__init__.py b/src/zenml/integrations/numpy/__init__.py index 9e81d58edcf..614797e032d 100644 --- a/src/zenml/integrations/numpy/__init__.py +++ b/src/zenml/integrations/numpy/__init__.py @@ -21,7 +21,7 @@ class NumpyIntegration(Integration): """Definition of Numpy integration for ZenML.""" NAME = NUMPY - REQUIREMENTS = ["numpy<2.0.0"] + REQUIREMENTS = ["numpy"] @classmethod def activate(cls) -> None: From f7fcadb09a0e49ae742f3e1245189e2377f94dd7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bar=C4=B1=C5=9F=20Can=20Durak?= <36421093+bcdurak@users.noreply.github.com> Date: Mon, 23 Sep 2024 17:05:39 +0200 Subject: [PATCH 2/2] Update src/zenml/integrations/numpy/__init__.py Co-authored-by: Hamza Tahir --- src/zenml/integrations/numpy/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/zenml/integrations/numpy/__init__.py b/src/zenml/integrations/numpy/__init__.py index 614797e032d..c10928e5b46 100644 --- a/src/zenml/integrations/numpy/__init__.py +++ b/src/zenml/integrations/numpy/__init__.py @@ -21,7 +21,7 @@ class NumpyIntegration(Integration): """Definition of Numpy integration for ZenML.""" NAME = NUMPY - REQUIREMENTS = ["numpy"] + REQUIREMENTS = ["numpy<3.0.0"] @classmethod def activate(cls) -> None: