zenml-nightly 0.70.0.dev20241119__py3-none-any.whl → 0.70.0.dev20241120__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.70.0.dev20241119
+0.70.0.dev20241120

zenml/service_connectors/service_connector_registry.py

@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 """Implementation of a service connector registry."""
 
+import threading
 from typing import TYPE_CHECKING, Dict, List, Optional, Union
 
 from zenml.logger import get_logger
@@ -34,6 +35,7 @@ class ServiceConnectorRegistry:
         """Initialize the service connector registry."""
         self.service_connector_types: Dict[str, ServiceConnectorTypeModel] = {}
         self.initialized = False
+        self.lock = threading.RLock()
 
     def register_service_connector_type(
         self,
@@ -46,23 +48,25 @@ class ServiceConnectorRegistry:
             service_connector_type: Service connector type.
             overwrite: Whether to overwrite an existing service connector type.
         """
-        if (
-            service_connector_type.connector_type
-            not in self.service_connector_types
-            or overwrite
-        ):
-            self.service_connector_types[
+        with self.lock:
+            if (
                 service_connector_type.connector_type
-            ] = service_connector_type
-            logger.debug(
-                "Registered service connector type "
-                f"{service_connector_type.connector_type}."
-            )
-        else:
-            logger.debug(
-                f"Found existing service connector for type "
-                f"{service_connector_type.connector_type}: Skipping registration."
-            )
+                not in self.service_connector_types
+                or overwrite
+            ):
+                self.service_connector_types[
+                    service_connector_type.connector_type
+                ] = service_connector_type
+                logger.debug(
+                    "Registered service connector type "
+                    f"{service_connector_type.connector_type}."
+                )
+            else:
+                logger.debug(
+                    f"Found existing service connector for type "
+                    f"{service_connector_type.connector_type}: Skipping "
+                    "registration."
+                )
 
     def get_service_connector_type(
         self,
@@ -201,54 +205,61 @@ class ServiceConnectorRegistry:
     def register_builtin_service_connectors(self) -> None:
         """Registers the default built-in service connectors."""
         # Only register built-in service connectors once
-        if self.initialized:
-            return
+        with self.lock:
+            if self.initialized:
+                return
 
-        self.initialized = True
+            self.initialized = True
 
-        try:
-            from zenml.integrations.aws.service_connectors.aws_service_connector import (  # noqa
-                AWSServiceConnector,
-            )
-        except ImportError as e:
-            logger.warning(f"Could not import AWS service connector: {e}.")
+            try:
+                from zenml.integrations.aws.service_connectors.aws_service_connector import (  # noqa
+                    AWSServiceConnector,
+                )
+            except ImportError as e:
+                logger.warning(f"Could not import AWS service connector: {e}.")
 
-        try:
-            from zenml.integrations.gcp.service_connectors.gcp_service_connector import (  # noqa
-                GCPServiceConnector,
-            )
-        except ImportError as e:
-            logger.warning(f"Could not import GCP service connector: {e}.")
+            try:
+                from zenml.integrations.gcp.service_connectors.gcp_service_connector import (  # noqa
+                    GCPServiceConnector,
+                )
+            except ImportError as e:
+                logger.warning(f"Could not import GCP service connector: {e}.")
 
-        try:
-            from zenml.integrations.azure.service_connectors.azure_service_connector import (  # noqa
-                AzureServiceConnector,
-            )
-        except ImportError as e:
-            logger.warning(f"Could not import Azure service connector: {e}.")
+            try:
+                from zenml.integrations.azure.service_connectors.azure_service_connector import (  # noqa
+                    AzureServiceConnector,
+                )
+            except ImportError as e:
+                logger.warning(
+                    f"Could not import Azure service connector: {e}."
+                )
 
-        try:
-            from zenml.integrations.kubernetes.service_connectors.kubernetes_service_connector import (  # noqa
-                KubernetesServiceConnector,
-            )
-        except ImportError as e:
-            logger.warning(
-                f"Could not import Kubernetes service connector: {e}."
-            )
+            try:
+                from zenml.integrations.kubernetes.service_connectors.kubernetes_service_connector import (  # noqa
+                    KubernetesServiceConnector,
+                )
+            except ImportError as e:
+                logger.warning(
+                    f"Could not import Kubernetes service connector: {e}."
+                )
 
-        try:
-            from zenml.service_connectors.docker_service_connector import (  # noqa
-                DockerServiceConnector,
-            )
-        except ImportError as e:
-            logger.warning(f"Could not import Docker service connector: {e}.")
+            try:
+                from zenml.service_connectors.docker_service_connector import (  # noqa
+                    DockerServiceConnector,
+                )
+            except ImportError as e:
+                logger.warning(
+                    f"Could not import Docker service connector: {e}."
+                )
 
-        try:
-            from zenml.integrations.hyperai.service_connectors.hyperai_service_connector import (  # noqa
-                HyperAIServiceConnector,
-            )
-        except ImportError as e:
-            logger.warning(f"Could not import HyperAI service connector: {e}.")
+            try:
+                from zenml.integrations.hyperai.service_connectors.hyperai_service_connector import (  # noqa
+                    HyperAIServiceConnector,
+                )
+            except ImportError as e:
+                logger.warning(
+                    f"Could not import HyperAI service connector: {e}."
+                )
 
 
 service_connector_registry = ServiceConnectorRegistry()

{zenml_nightly-0.70.0.dev20241119.dist-info → zenml_nightly-0.70.0.dev20241120.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenml-nightly
-Version: 0.70.0.dev20241119
+Version: 0.70.0.dev20241120
 Summary: ZenML: Write production-ready ML code.
 Home-page: https://zenml.io
 License: Apache-2.0

{zenml_nightly-0.70.0.dev20241119.dist-info → zenml_nightly-0.70.0.dev20241120.dist-info}/RECORD

@@ -6,7 +6,7 @@ RELEASE_NOTES.md,sha256=DleauURHESDrTrcVzCVLqPiSM9NIAk5vldvEFMc7qlk,389375
 ROADMAP.md,sha256=hiLSmr16BH8Dfx7SaQM4JcXCGCVl6mFZPFAwJeDTrJU,407
 SECURITY.md,sha256=9DepA8y03yvCZLHEfcXLTDH4lUyKHquAdukBsccNN7c,682
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=RmxPlX0kaYoYYqigOE1L6G416EVE62myhwmKkfCvdCA,19
+zenml/VERSION,sha256=hmQd5okP_IQNYZbqN_n_-p28vX5eUklJ2KovkmOM5Qg,19
 zenml/__init__.py,sha256=SkMObQA41ajqdZqGErN00S1Vf3KAxpLvbZ-OBy5uYoo,2130
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -705,7 +705,7 @@ zenml/secret/schemas/gcp_secret_schema.py,sha256=tDSxPXlfh4BOSICZjfJjUukfzEyBEgx
 zenml/service_connectors/__init__.py,sha256=gIBAVUPBZtdO6JtEBCixy9YG4wZRA1mnaxaGAxi3T1Q,645
 zenml/service_connectors/docker_service_connector.py,sha256=_6WPqYCcSUf8JPakipbkJRvaN2ghhY7zCr38WCHG0SI,13218
 zenml/service_connectors/service_connector.py,sha256=HW8dTE826ZSD1vWQlqQRm8XYq08v7tDZQ8ORnJ5rTcE,55351
-zenml/service_connectors/service_connector_registry.py,sha256=lP0s0k7RIp9WZEW36Q0ygHD5urzF-wUwh41rbXWS5ss,9186
+zenml/service_connectors/service_connector_registry.py,sha256=mCabyKAr7Y6bcAbIFXbir9YrHczDe1ZJ8Bks5bOQouk,9658
 zenml/service_connectors/service_connector_utils.py,sha256=h3wfVSPO0sQnARVRnih3l8P7LwkCmAgijThuHp2Z53E,18026
 zenml/services/__init__.py,sha256=8aSLtYlX9EVLB1Se1I4dbhZdTqrjNrAfsn1d_RCc_nk,2800
 zenml/services/container/__init__.py,sha256=dFHcmlXgNXjUaCR18oGQJ19-I_6f3UeAUURHjqldjTs,668
@@ -1268,8 +1268,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Bq1djrUP9saoD7vECjS7
 zenml/zen_stores/sql_zen_store.py,sha256=v9aIoBght-Pjivr2OYxvANBOhSqx8cUOzZh0G_oJOnw,404327
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.70.0.dev20241119.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.70.0.dev20241119.dist-info/METADATA,sha256=FeRnTMQEqSj5-Yhqg1aXhzsMCt72lm2v18S6_N4bFeA,21208
-zenml_nightly-0.70.0.dev20241119.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-zenml_nightly-0.70.0.dev20241119.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.70.0.dev20241119.dist-info/RECORD,,
+zenml_nightly-0.70.0.dev20241120.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.70.0.dev20241120.dist-info/METADATA,sha256=1ESYfHisI0IYMmkzwGoPrO2mhYcBpka_v3zpzog-Km4,21208
+zenml_nightly-0.70.0.dev20241120.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+zenml_nightly-0.70.0.dev20241120.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.70.0.dev20241120.dist-info/RECORD,,