clarifai 11.3.0rc2__py3-none-any.whl → 11.4.0__py3-none-any.whl

clarifai/client/deployment.py

@@ -7,60 +7,65 @@ from clarifai.utils.protobuf import dict_to_protobuf
 
 
 class Deployment(Lister, BaseClient):
-  """Deployment is a class that provides access to Clarifai API endpoints related to Deployment information."""
+    """Deployment is a class that provides access to Clarifai API endpoints related to Deployment information."""
 
-  def __init__(self,
-               deployment_id: str = None,
-               user_id: str = None,
-               base_url: str = "https://api.clarifai.com",
-               pat: str = None,
-               token: str = None,
-               root_certificates_path: str = None,
-               **kwargs):
-    """Initializes a Deployment object.
-
-    Args:
-        deployment_id (str): The Deployment ID for the Deployment to interact with.
-        user_id (str): The user ID of the user.
-        base_url (str): Base API url. Default "https://api.clarifai.com"
-        pat (str): A personal access token for authentication. Can be set as env var CLARIFAI_PAT
-        token (str): A session token for authentication. Accepts either a session token or a pat. Can be set as env var CLARIFAI_SESSION_TOKEN
-        root_certificates_path (str): Path to the SSL root certificates file, used to establish secure gRPC connections.
-        **kwargs: Additional keyword arguments to be passed to the deployment.
-    """
-    self.kwargs = {**kwargs, 'id': deployment_id, 'user_id': user_id}
-    self.deployment_info = resources_pb2.Deployment()
-    dict_to_protobuf(self.deployment_info, self.kwargs)
-    self.logger = logger
-    BaseClient.__init__(
+    def __init__(
         self,
-        user_id=user_id,
-        base=base_url,
-        pat=pat,
-        token=token,
-        root_certificates_path=root_certificates_path)
-    Lister.__init__(self)
+        deployment_id: str = None,
+        user_id: str = None,
+        base_url: str = "https://api.clarifai.com",
+        pat: str = None,
+        token: str = None,
+        root_certificates_path: str = None,
+        **kwargs,
+    ):
+        """Initializes a Deployment object.
+
+        Args:
+            deployment_id (str): The Deployment ID for the Deployment to interact with.
+            user_id (str): The user ID of the user.
+            base_url (str): Base API url. Default "https://api.clarifai.com"
+            pat (str): A personal access token for authentication. Can be set as env var CLARIFAI_PAT
+            token (str): A session token for authentication. Accepts either a session token or a pat. Can be set as env var CLARIFAI_SESSION_TOKEN
+            root_certificates_path (str): Path to the SSL root certificates file, used to establish secure gRPC connections.
+            **kwargs: Additional keyword arguments to be passed to the deployment.
+        """
+        self.kwargs = {**kwargs, 'id': deployment_id, 'user_id': user_id}
+        self.deployment_info = resources_pb2.Deployment()
+        dict_to_protobuf(self.deployment_info, self.kwargs)
+        self.logger = logger
+        BaseClient.__init__(
+            self,
+            user_id=user_id,
+            base=base_url,
+            pat=pat,
+            token=token,
+            root_certificates_path=root_certificates_path,
+        )
+        Lister.__init__(self)
 
-  @staticmethod
-  def get_runner_selector(user_id: str, deployment_id: str) -> resources_pb2.RunnerSelector:
-    """Returns a RunnerSelector object for the given deployment_id.
+    @staticmethod
+    def get_runner_selector(user_id: str, deployment_id: str) -> resources_pb2.RunnerSelector:
+        """Returns a RunnerSelector object for the given deployment_id.
 
-    Args:
-        deployment_id (str): The deployment ID for the deployment.
+        Args:
+            deployment_id (str): The deployment ID for the deployment.
 
-    Returns:
-        resources_pb2.RunnerSelector: A RunnerSelector object for the given deployment_id.
-    """
-    return resources_pb2.RunnerSelector(
-        deployment=resources_pb2.Deployment(id=deployment_id, user_id=user_id))
+        Returns:
+            resources_pb2.RunnerSelector: A RunnerSelector object for the given deployment_id.
+        """
+        return resources_pb2.RunnerSelector(
+            deployment=resources_pb2.Deployment(id=deployment_id, user_id=user_id)
+        )
 
-  def __getattr__(self, name):
-    return getattr(self.deployment_info, name)
+    def __getattr__(self, name):
+        return getattr(self.deployment_info, name)
 
-  def __str__(self):
-    init_params = [param for param in self.kwargs.keys()]
-    attribute_strings = [
-        f"{param}={getattr(self.deployment_info, param)}" for param in init_params
-        if hasattr(self.deployment_info, param)
-    ]
-    return f"Deployment Details: \n{', '.join(attribute_strings)}\n"
+    def __str__(self):
+        init_params = [param for param in self.kwargs.keys()]
+        attribute_strings = [
+            f"{param}={getattr(self.deployment_info, param)}"
+            for param in init_params
+            if hasattr(self.deployment_info, param)
+        ]
+        return f"Deployment Details: \n{', '.join(attribute_strings)}\n"