Skip to content

Commit 80c33d6

Browse files
strickvladtygan
authored andcommitted
Relax azure, gcfs and s3 dependencies (zenml-io#2498)
1 parent c04f210 commit 80c33d6

File tree

4 files changed

+11
-7
lines changed

4 files changed

+11
-7
lines changed

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ jinja2-time = { version = "^0.2.0", optional = true }
8181
python-terraform = { version = "^0.10.1", optional = true }
8282

8383
# Optional dependencies for the AWS secrets store
84-
boto3 = { version = ">=1.16.0,<=1.24.59", optional = true }
84+
boto3 = { version = ">=1.16.0", optional = true }
8585

8686
# Optional dependencies for the GCP secrets store
8787
google-cloud-secret-manager = { version = ">=2.12.5", optional = true }
@@ -111,10 +111,10 @@ azure-storage-blob = { version = ">=12.0.0", optional = true }
111111
azure-mgmt-resource = { version = ">=21.0.0", optional = true }
112112

113113
# Optional dependencies for the S3 artifact store
114-
s3fs = { version = "2022.11.0", optional = true }
114+
s3fs = { version = ">=2022.11.0", optional = true }
115115

116116
# Optional dependencies for the GCS artifact store
117-
gcsfs = { version = "2022.11.0", optional = true }
117+
gcsfs = { version = ">=2022.11.0", optional = true }
118118

119119
# Optional dependencies for the Azure artifact store
120120
adlfs = { version = ">=2021.10.0", optional = true }

src/zenml/integrations/azure/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ class AzureIntegration(Integration):
3939

4040
NAME = AZURE
4141
REQUIREMENTS = [
42-
"adlfs==2021.10.0",
42+
"adlfs>=2021.10.0",
4343
"azure-keyvault-keys",
4444
"azure-keyvault-secrets",
4545
"azure-identity==1.10.0",

src/zenml/integrations/huggingface/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,11 @@ class HuggingfaceIntegration(Integration):
2121
"""Definition of Huggingface integration for ZenML."""
2222

2323
NAME = HUGGINGFACE
24-
REQUIREMENTS = ["transformers<=4.31", "datasets"]
24+
REQUIREMENTS = [
25+
"transformers<=4.31",
26+
"datasets",
27+
"huggingface_hub>0.19.0",
28+
]
2529

2630
@classmethod
2731
def activate(cls) -> None:

src/zenml/integrations/s3/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@ class S3Integration(Integration):
4141
# The above command installs boto3==1.26.76, so we use the same version
4242
# here to avoid the dependency resolution overhead.
4343
REQUIREMENTS = [
44-
"s3fs>2022.3.0,<=2023.4.0",
45-
"boto3<=1.26.76",
44+
"s3fs>2022.3.0",
45+
"boto3",
4646
# The following dependencies are only required for the AWS connector.
4747
"aws-profile-manager",
4848
]

0 commit comments

Comments
 (0)