def get_pod_template_spec(self, resource_name, volume_mounts, volumes, labels, env_vars=None, command=None, args=None, init_command=None, init_args=None, init_env_vars=None, ports=None, persistence_outputs=None, persistence_data=None, outputs_refs_jobs=None, outputs_refs_experiments=None, secret_refs=None, configmap_refs=None, resources=None, ephemeral_token=None, node_selector=None, affinity=None, tolerations=None, restart_policy=None, init_context_mounts=None, sidecar_context_mounts=None): annotations = None if requests_tpu(resources): annotations = get_tpu_annotations() metadata = client.V1ObjectMeta(name=resource_name, labels=labels, namespace=self.namespace, annotations=annotations) pod_spec = self.get_task_pod_spec( resource_name=resource_name, volume_mounts=volume_mounts, volumes=volumes, env_vars=env_vars, command=command, args=args, init_command=init_command, init_args=init_args, init_env_vars=init_env_vars, ports=ports, persistence_outputs=persistence_outputs, persistence_data=persistence_data, outputs_refs_jobs=outputs_refs_jobs, outputs_refs_experiments=outputs_refs_experiments, secret_refs=secret_refs, configmap_refs=configmap_refs, resources=resources, ephemeral_token=ephemeral_token, node_selector=node_selector, affinity=affinity, tolerations=tolerations, init_context_mounts=init_context_mounts, sidecar_context_mounts=sidecar_context_mounts, restart_policy=restart_policy) return client.V1PodTemplateSpec(metadata=metadata, spec=pod_spec)
def get_pod_spec(self, resource_name, volume_mounts, volumes, labels, env_vars=None, command=None, args=None, init_command=None, init_args=None, init_env_vars=None, ports=None, persistence_outputs=None, persistence_data=None, outputs_refs_jobs=None, outputs_refs_experiments=None, secret_refs=None, config_map_refs=None, resources=None, annotations=None, ephemeral_token=None, node_selector=None, affinity=None, tolerations=None, max_restarts=None, restart_policy=None, reconcile_url=None, init_context_mounts=None, sidecar_context_mounts=None): resources = self._get_pod_resources(resources=resources) resources_annotations = {} annotations = self._get_annotations(annotations=annotations) if requests_tpu(resources): resources_annotations = get_tpu_annotations() if annotations: resources_annotations.update(annotations) annotations = resources_annotations metadata = client.V1ObjectMeta(name=resource_name, labels=labels, namespace=self.namespace, annotations=annotations) pod_spec = self.get_task_pod_spec( resource_name=resource_name, volume_mounts=volume_mounts, volumes=volumes, env_vars=env_vars, command=command, args=args, init_command=init_command, init_args=init_args, init_env_vars=init_env_vars, ports=ports, persistence_outputs=persistence_outputs, persistence_data=persistence_data, outputs_refs_jobs=outputs_refs_jobs, outputs_refs_experiments=outputs_refs_experiments, secret_refs=secret_refs, config_map_refs=config_map_refs, resources=resources, ephemeral_token=ephemeral_token, node_selector=node_selector, affinity=affinity, tolerations=tolerations, init_context_mounts=init_context_mounts, sidecar_context_mounts=sidecar_context_mounts, reconcile_url=reconcile_url, max_restarts=max_restarts, restart_policy=restart_policy) return metadata, pod_spec