zenml-nightly 0.81.0.dev20250417__py3-none-any.whl → 0.81.0.dev20250419__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.81.0.dev20250417
+0.81.0.dev20250419

zenml/integrations/github/code_repositories/github_code_repository.py

@@ -16,6 +16,7 @@
 import os
 import re
 from typing import Any, Dict, List, Optional
+from urllib.parse import urlparse
 from uuid import uuid4
 
 import requests
@@ -237,19 +238,37 @@ class GitHubCodeRepository(BaseCodeRepository):
         Returns:
             Whether the remote url is correct.
         """
+        # Normalize repository information
         host = self.config.host or "github.com"
         host = host.rstrip("/")
-
-        https_url = (
-            f"https://{host}/{self.config.owner}/{self.config.repository}.git"
-        )
-        if url == https_url:
-            return True
-
-        ssh_regex = re.compile(
-            f".*@{host}:{self.config.owner}/{self.config.repository}.git"
-        )
-        if ssh_regex.fullmatch(url):
-            return True
+        owner = self.config.owner
+        repo = self.config.repository
+
+        # Clean the input URL by removing any trailing slashes
+        url = url.rstrip("/")
+
+        # Handle HTTPS URLs using urlparse
+        parsed_url = urlparse(url)
+        if parsed_url.scheme == "https":
+            expected_path = f"/{owner}/{repo}"
+            actual_path = parsed_url.path.removesuffix(".git")
+            return parsed_url.hostname == host and actual_path == expected_path
+
+        # Create regex patterns for non-HTTPS URL formats
+        patterns = [
+            # SSH format: git@github.com:owner/repo[.git]
+            rf"^[^@]+@{re.escape(host)}:{re.escape(owner)}/{re.escape(repo)}(\.git)?$",
+            # Alternative SSH: ssh://git@github.com/owner/repo[.git]
+            rf"^ssh://[^@]+@{re.escape(host)}/{re.escape(owner)}/{re.escape(repo)}(\.git)?$",
+            # Git protocol: git://github.com/owner/repo[.git]
+            rf"^git://{re.escape(host)}/{re.escape(owner)}/{re.escape(repo)}(\.git)?$",
+            # GitHub CLI: gh:owner/repo
+            rf"^gh:{re.escape(owner)}/{re.escape(repo)}$",
+        ]
+
+        # Try matching against each pattern
+        for pattern in patterns:
+            if re.match(pattern, url):
+                return True
 
         return False

zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py

@@ -178,23 +178,25 @@ class GitLabCodeRepository(BaseCodeRepository):
         """
         host = self.config.host or "gitlab.com"
         host = host.rstrip("/")
+        group = self.config.group
+        project = self.config.project
 
+        # Handle HTTPS URLs
         parsed_url = urlparse(url)
-        if (
-            parsed_url.scheme == "https"
-            and parsed_url.hostname == host
-            and parsed_url.path
-            == f"/{self.config.group}/{self.config.project}.git"
-        ):
-            return True
+        if parsed_url.scheme == "https" and parsed_url.hostname == host:
+            # Remove .git suffix if present for comparison
+            expected_path = f"/{group}/{project}"
+            actual_path = parsed_url.path.removesuffix(".git")
+            return actual_path == expected_path
 
+        # Handle SSH URLs
         ssh_regex = re.compile(
             r"^(?P<scheme_with_delimiter>ssh://)?"
             r"(?P<userinfo>git)"
             f"@{host}:"
             r"(?P<port>\d+)?"
             r"(?(scheme_with_delimiter)/|/?)"
-            f"{self.config.group}/{self.config.project}.git$",
+            f"{group}/{project}(\.git)?$",
         )
         if ssh_regex.fullmatch(url):
             return True

{zenml_nightly-0.81.0.dev20250417.dist-info → zenml_nightly-0.81.0.dev20250419.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.81.0.dev20250417
+Version: 0.81.0.dev20250419
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.81.0.dev20250417.dist-info → zenml_nightly-0.81.0.dev20250419.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=aYtF-ksj5iJaP_lXns-UKRyoEI5LJOygqNJdcYVaIzY,19
+zenml/VERSION,sha256=FmOrvF3wrz-iTCu4wD0WjP-UDy417cFyDc02QOsvjqo,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -277,14 +277,14 @@ zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=X8CCniyAo7N
 zenml/integrations/gcp/vertex_custom_job_parameters.py,sha256=B5RLkw7KDOi4ZfWHFnC6TGLphXMzToMjROxszCEAS9c,3676
 zenml/integrations/github/__init__.py,sha256=movoEHoX8SpY0t6mBS5D64enxG6lPkpYJIEQ9jElHLA,1427
 zenml/integrations/github/code_repositories/__init__.py,sha256=ub_hSE2ks2mZB1aeHRjQYz7QIRQIgOw2s080IIqJaGs,817
-zenml/integrations/github/code_repositories/github_code_repository.py,sha256=1tkDQNK7X87bf5E2pdA9Ll49RksY1hc2lcS9h5cNDfI,8392
+zenml/integrations/github/code_repositories/github_code_repository.py,sha256=wrx-orSgEfEsicFNwHXfJ27rn5v0nQE1SSOsBuC_Ch8,9395
 zenml/integrations/github/plugins/__init__.py,sha256=yf7xkBs8wEUMP2-nFbDIVeXs1omHtZoyZBgobMYB1l0,804
 zenml/integrations/github/plugins/event_sources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/integrations/github/plugins/event_sources/github_webhook_event_source.py,sha256=nU--Wk9s2W7koF-JrtWIrSpuNWNVouftHQDiEY8dkdQ,17101
 zenml/integrations/github/plugins/github_webhook_event_source_flavor.py,sha256=jth8sxrmyg22-wT5Ax0fdsiLhTQwHXxaiTnB3kD97pk,1669
 zenml/integrations/gitlab/__init__.py,sha256=dxo7rxGj-w8ZLgjx9xlDGOSBfGmQylUAROmeabQkUp8,964
 zenml/integrations/gitlab/code_repositories/__init__.py,sha256=Ds7NL6tCqLApRsOgvUofEq3Ms2No5_Z095uvi1gLVIk,817
-zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py,sha256=QLiDTHHZBblpWg1y6DmV2bsuxFs5FNOo9E903yLlGbs,6532
+zenml/integrations/gitlab/code_repositories/gitlab_code_repository.py,sha256=miWGRNjIny5ZvLaFsrFUwFSzdQ3-xPyND9vuI3EyrKE,6701
 zenml/integrations/great_expectations/__init__.py,sha256=x6L4OokAE_Q0sC1vgZGi1X705AJIe_TNpdAEVKAvgFU,2546
 zenml/integrations/great_expectations/data_validators/__init__.py,sha256=Z16qmLfUoataEABQ6Ec-HSLM_a9VRALHFa4OoAyozIk,857
 zenml/integrations/great_expectations/data_validators/ge_data_validator.py,sha256=qp2ZFqQiYPszRc6vGhZhK22GEHhGoTQ0Y9u0trXNQyg,21404
@@ -1316,8 +1316,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=7ISpj0CC31xb66ehW1PZZ2Yapk3Sxu-CiuOGZlLqGQA,441658
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.81.0.dev20250417.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.81.0.dev20250417.dist-info/METADATA,sha256=QeRiC-2To8f8FbrB4LJSeMr51h2RZtG82F_nBhWIOrY,24312
-zenml_nightly-0.81.0.dev20250417.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-zenml_nightly-0.81.0.dev20250417.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.81.0.dev20250417.dist-info/RECORD,,
+zenml_nightly-0.81.0.dev20250419.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.81.0.dev20250419.dist-info/METADATA,sha256=rAyG36q25Lse3NGIagH-PcQOpIN7lkAW2h7esLrqRws,24312
+zenml_nightly-0.81.0.dev20250419.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.81.0.dev20250419.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.81.0.dev20250419.dist-info/RECORD,,